代码随想Day23 | 669. 修剪二叉搜索树、108.将有序数组转换为二叉搜索树、538.把二叉搜索树转换为累加树

669. 修剪二叉搜索树 

这道题目的难点在于，并不是简单的判断当前节点的值就可以，如果当前节点不满足条件，需要根据大小关系 ，递归找到当前节点满足条件的子树，并连接到该节点的父节点上；另外这一题同样运用了当前节点被上一层接住的思想，详细代码如下：

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        //trim BST
        if (root == nullptr) return root;
        if (root->val > high) return trimBST(root->left, low, high); // find left child in [low,high]
        if (root->val < low) return trimBST(root->right, low, high); //find right child in [low, high]
        root->left = trimBST(root->left, low, high);    //root->left接入符合条件的左孩子
        root->right = trimBST(root->right, low, high);  //root->right接入符合条件的右孩子
        return root;
    }
};

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

这道题目是构建二叉树，类似于之前的根据前序和中序构造二叉树的思想，但是这道题更加简单，主要的思想是构建二叉搜索树，同时还要是平衡二叉树，因此只需要每次把数组最中间的树当作根节点，左边的当作左子树，右边的当作右子树就可以构造出平衡二叉树，详细代码如下：

class Solution
{
public:
    TreeNode* dfs(vector<int>& nums, int l, int r) 
    {
        //左闭右开
        if (l >= r) return nullptr;
        //找中间值
        int index = l + (r - l) / 2;
        TreeNode* root = new TreeNode(nums[index]);
        if (r - l == 1) return root; //只有一个元素，直接返回
        //分割
        root->left = dfs(nums, l, index);
        root->right = dfs(nums, index + 1, r);
        return root;
    }

    TreeNode* sortedArrayToBST(vector<int>& nums) 
    {
        if (nums.size() == 0) return nullptr;
        return dfs(nums, 0, nums.size());
    }
};

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

这道题只需要用中序遍历和指针思想即可解决，不过要保持“右中左”，因为累加是倒序的；另外使用pre指针来保存上一个节点，这样就可完成累加，详细代码如下：

class Solution
{
public:
    TreeNode* pre = nullptr;
    void dfs(TreeNode* root) 
    {
        if (root == nullptr) return ;
        dfs(root->right); //right
        //mid
        if (pre != nullptr) root->val += pre->val;
        pre = root;
        dfs(root->left); //left

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