代码随想录算法训练营第十九天| 二叉树 669. 修剪二叉搜索树 108.确定递归函数返回值及其参数

发布时间:2024年01月05日

669. 修剪二叉搜索树

修剪二叉树通过设置root将左右结点的值接住,如果不符合,则根据二叉搜索树特点重新搜索

确定递归函数的参数以及返回值:返回值为结点,参数为root,边界

确定终止条件:修剪的操作并不是在终止条件上进行的,所以就是遇到空节点返回就可以了。

确定单层逻辑:如果root(当前节点)的元素小于low的数值,那么应该递归右子树,并返回右子树符合条件的头结点。如果root(当前节点)的元素大于high的,那么应该递归左子树,并返回左子树符合条件的头结点。接下来要将下一层处理完左子树的结果赋给root->left,处理完右子树的结果赋给root->right。

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(root==NULL)return NULL;
        if(root->val<low){
            //注意审题,题目是二叉搜索树,如果结点值比最小值小,则它的右子树有可能会在范围内
            TreeNode* right=trimBST(root->right,low,high);
            return right;
        }
        if(root->val>high){
            TreeNode* left=trimBST(root->left,low,high);
            return left;
        }
        root->left=trimBST(root->left,low,high);
        root->right=trimBST(root->right,low,high);
        return root;
    }
};

108. 将有序数组转换为二叉搜索树

构造二叉树的题目都可以用前序遍历来做,因此需要找到前序遍历的根结点,即中间的位置。本题参看前面构造最大二叉树的方法,采用前序遍历的方法,为了让node不被size为1时不存在,在最开始首先定义node,然后找出中间结点。

确定递归函数返回值及其参数:参数为数组,返回值为结点

确定终止条件:如果数组只剩1个或者没有了,需要返回

确定递归逻辑:首先先找到中间结点,对于双数的数组来说,中间结点可以是左边也可以是右边,这里选择左边的(为了后续正确往右遍历而不会溢出)。然后将左结点接住左边的数组,右结点接住右边的数组即可。

class Solution {
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        TreeNode* node=new TreeNode();
        if(nums.size()==1){
            node->val=nums[0];
            return node;
        }
        if(nums.size()==0)return NULL;
        int mid=(nums.size()-1)/2;
        node->val=nums[mid];
        vector<int> left(nums.begin(),nums.begin()+mid);
        vector<int> right(nums.begin()+mid+1,nums.end());
        node->left=sortedArrayToBST(left);
        node->right=sortedArrayToBST(right);
        return node;
    }
};

538. 把二叉搜索树转换为累加树

本题只需要理解题目就挺容易的,题目的意思为新二叉树上面的结点大小等于原二叉树比它大的所有结点的和。由于原二叉树是一个二叉搜索树,最大的值在最右下角的结点,可以从最大的开始遍历,因此本题的遍历顺序应该为右中左。

确定递归函数参数以及返回值:不需要递归函数的返回值做什么操作了,要遍历整棵树。同时需要定义一个全局变量pre,用来保存cur节点的前一个节点的数值

确定终止条件:空结点终止,表示到底了

确定单层递归逻辑:中节点的处理逻辑就是让cur的数值加上前一个节点的数值。

class Solution {
public:
    TreeNode* pre=NULL;
    void traversal(TreeNode* cur){
        if(cur==NULL)return ;
        traversal(cur->right);
        if(pre)cur->val+=pre->val;
        pre=cur;
        traversal(cur->left);
    }
    TreeNode* convertBST(TreeNode* root) {
        traversal(root);
        return root;
    }
};

文章来源:https://blog.csdn.net/weixin_47100514/article/details/135380306
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。