代码随想录算法训练营第21天| 669.修剪二叉搜索树、108 .将有序数组转换为二叉搜索树 、538 .把二叉搜索树转换为累加树

669.修剪二叉搜索树

题目链接：669. 修剪二叉搜索树

思想：这道题我采用了从底部向上遍历的方法，当一个节点不满足条件时，就直接返回空指针或他的右子树或者左子树。由于是自底向上遍历，所以不担心符合条件的节点会被误删，然后把符合条件的节点返回给上一层就可以。除此之外，采用先序遍历也可以，如果当前节点不符合条件，就对他的左子树或者右子树中符合条件的节点进行相同的处理，然后返回给上一层节点。

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(root==nullptr) return nullptr;
        //如果当前结点小于low 返回他符合条件的第一个右子树
        root->left = trimBST(root->left,low,high);
        root->right = trimBST(root->right,low,high);
        
        if(root->val<low){
            while(root->right){
                if(root->right->val>=low) break;
                else root = root->right;
            }
            return root->right;
        }
        //如果当前结点大于high 返回它符合条件的第一个左子树
        else if(root->val>high){
            while(root->left){
                if(root->left->val<=high) break;
                else root = root->left;
            }
            return root->left;
        }
        
        return root;
    }
};

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

题目链接：108. 将有序数组转换为二叉搜索树

思想：因为是平衡二叉搜索树所以要左右子树的元素个数相差不可以超过1，找到数组中最终间的元素并将其作为根节点，通过元素下标将原数组分为左右两个部分，对两个新数组分别进行上面相同的操作，并将结果向上一层返回。

class Solution {
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        if(nums.size()==0) return nullptr;      
        int mid = nums.size()/2;
        TreeNode* root = new TreeNode();
        root->val = nums[mid];
        if(nums.size()==1) return root;

        vector<int> newleft(nums.begin(),nums.begin()+mid);
        vector<int> newright(nums.begin()+mid+1,nums.end());
        root->left = sortedArrayToBST(newleft);
        root->right = sortedArrayToBST(newright);
        return root;
    }
};

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

题目链接：538. 把二叉搜索树转换为累加树

思想：右根左进行遍历，设置一个sum记录总和，当访问到一个结点的时候，更新sum=sum+当前val，然后把新的值赋给当前节点。

class Solution {
public:
int sum = 0;
    TreeNode* convertBST(TreeNode* root) {
        if(root==nullptr)return nullptr;
        
        root->right = convertBST(root->right);
        sum+=root->val;
        root->val = sum;

        root->left = convertBST(root->left);
        return root;
    }
};