【代码随想录算法训练Day23】LeetCode 669. 修剪二叉搜索树、LeetCode 108.将有序数组转换为二叉搜索树、LeetCode 538.把二叉搜索树转换为累加树

Day23 二叉树第九天

LeetCode 669. 修剪二叉搜索树

这道题的难点在于理解做法，代码看起来倒是很简单，但是很难理解。
如果有一个节点在范围之外，大于范围就用左子树修剪的结果替代这个节点，否则用左子树修剪的结果替代这个节点，然后递归修剪左右子树。

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(!root) return NULL;
        if(root->val < low)
            return trimBST(root->right,low,high);
        if(root->val > high)
            return trimBST(root->left,low,high);
        root->left=trimBST(root->left,low,high);
        root->right=trimBST(root->right,low,high);
        return root;
    }
};

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

递归遍历建树哦，应该难不倒你了。
这道题的关键在于区间划分，固定左闭右闭即可。

class Solution {
public:
    TreeNode* traversal(vector<int>& nums,int left,int right){//左闭右闭
        if(left>right) return NULL;
        int mid=left+right>>1;
        TreeNode* root=new TreeNode(nums[mid]);
        root->left=traversal(nums,left,mid-1);
        root->right=traversal(nums,mid+1,right);
        return root;
    }

    TreeNode* sortedArrayToBST(vector<int>& nums) {
        return traversal(nums,0,nums.size()-1);
    }
};

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

又是一道遍历二叉树处理节点的问题。
老方法，先搞清楚线性递增数组中累加的逻辑：从最后一个元素开始一直逆着加到第一个数。
在这里也一样，中序遍历后是递增数组，那我们的操作逻辑就是中序数组倒过来，也就是右中左。
中的操作逻辑和之前双指针一样，用pre存储前一个节点的值，每次遍历加上pre即可，然后把pre更新成当前节点的值。

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

    TreeNode* convertBST(TreeNode* root) {
        traversal(root);
        return root;
    }
};

二叉树到这里就结束了，整整十天的陪伴，确实不容易，后面有时间会对二叉树做一个总结。