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

LeetCode 669.修剪二叉搜索树

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

踩坑：意识到了需要删除节点，但是没意识到要递归的删除。同时，与删除目标节点的删除逻辑也不完全相同。

思路：遍历二叉搜索树，遇到在区间之外的节点，若小于low则该节点的左子树肯定也都小于low所以只需要考虑右子树，大于high同理。这里如果按照删除目标节点的逻辑应该返回右（左）子树根节点，但是右（左）子树中依然有可能有超出区间的节点，所以要对子树进行递归，返回修剪后的子树根节点。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(!root) return nullptr;
        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.将有序数组转换为二叉搜索树

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

踩坑：没什么坑

思路：与构造二叉树类似，题目要求构造平衡二叉搜索树，只需要选择中间元素做根节点就可以保证左右子树节点个数平均。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        if(nums.empty()) return nullptr;
        
        TreeNode* root = new TreeNode(nums[nums.size()/2]);
        if(nums.size() == 1) return root;
        
        vector<int> left(nums.begin(), nums.begin() + nums.size()/2);
        vector<int> right(nums.begin() + nums.size()/2 + 1, nums.end());
        root->left = sortedArrayToBST(left);
        root->right = sortedArrayToBST(right);
        return root;
    }
};

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

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

踩坑：一开始想让函数返回累加值，但是发现这样反而在空节点时麻烦同时也没必要。因为要遍历整颗二叉树，一般不需要返回值。

思路：将大于等于当前节点的值累加赋给当前节点，我们知道左中右遍历二叉搜索树可以得到升序序列，那右中左就可以得到降序序列，从而也可以计算累加值。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void digui(TreeNode* root,int& sum)
    {
        if(!root) return;

        digui(root->right, sum);
        root->val = root->val + sum;
        sum = root->val;
        digui(root->left, sum);
    }
    TreeNode* convertBST(TreeNode* root) {
        int sum = 0;
        digui(root, sum);
        return root;
    }
};