【LeetCode】day21：669 - 修剪二叉搜索树, 108 - 将有序数组转换为二叉搜索树, 538 - 把二叉搜索树转换为累加树

669.修剪二叉搜索树

题目描述：

给你二叉搜索树的根节点 root ，同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树，使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即，如果没有被移除，原有的父代子代关系都应当保留)。 可以证明，存在 唯一的答案 。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意，根节点可能会根据给定的边界发生改变。

由于是二叉搜索树，剪枝情况可以简化：当前节点值小于边界low时，由于左子树中所有值均小于当前节点值，所以可以直接返回右子树；同理当前节点值大于边界high时，由于右子树中所有值均大于当前节点值，所以可以直接返回左子树。代码如下：

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if (root == nullptr) 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;
    }
};

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

题目描述：

给你一个整数数组 nums ，其中元素已经按 升序 排列，请你将其转换为一棵 平衡 二叉搜索树。

由于二叉搜索树序列化后的排列顺序为【左子树｜根节点｜右子树】，可先构造根节点，然后划分数组构建左右子树。这里使用二分的方式构造，可以使二叉树尽量平衡。代码如下：

class Solution {
private:
    TreeNode* traverse(vector<int>& nums, int left, int right) {
        if (left > right) return nullptr;
        int mid = (left + right) / 2;
        TreeNode* root = new TreeNode(nums[mid]);
        root->left = traverse(nums, left, mid - 1);
        root->right = traverse(nums, mid + 1, right);
        return root;
    }

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

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

题目描述：

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下，二叉搜索树满足下列约束条件：
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。

大于等于当前节点值的所有值之和，即为当前节点值与其右子树所有节点值、以及当前递归所累积的sum总和。可使用中序遍历的方式从右节点向左节点遍历，根据二叉搜索树的性质，每次遍历相当于是从升序序列的反向、即降序序列一次计算和，符合题意。代码如下：

class Solution {
private:
    int _sum = 0;

public:
    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;
    }
};