代码随想录 第二十二天 108.将有序数组转换为二叉搜索树||538.把二叉搜索树转换为累加树||

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

力扣题目链接(opens new window)

将一个按照升序排列的有序数组，转换为一棵高度平衡二叉搜索树。

本题中，一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

算法的基本思想是：

类似于二分法，将数组元素分成左右两部分，分别将左右两部分赋予构造的平衡BST的左孩子和右孩子。

代码如下：

class Solution {
private:
    TreeNode* BuildBST(vector<int>num,int left,int right) {
        if (left > right)return NULL;
        int mid = left + ((right - left) / 2);
        TreeNode*  root = new TreeNode(num[mid]);
        root->left = BuildBST(num, left, mid - 1);
        root->right = BuildBST(num, mid+1, right);
        return  root；  
   }
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        TreeNode* root = BuildBST(nums, 0, nums.size() - 1);
        return root;
    }
};

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

力扣题目链接(opens new window)

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键 小于 节点键的节点。 节点的右子树仅包含键 大于 节点键的节点。 左右子树也必须是二叉搜索树。

示例 1：

• 输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
• 输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

示例 2：

• 输入：root = [0,null,1]
• 输出：[1,null,1]

示例 3：

• 输入：root = [1,0,2]
• 输出：[3,3,2]

示例 4：

• 输入：root = [3,2,4,1]
• 输出：[7,9,4,10]

提示：

• 树中的节点数介于 0 和 104 之间。
• 每个节点的值介于 -104 和 104 之间。
• 树中的所有值 互不相同 。
• 给定的树为二叉搜索树。

核心思路：一开始我的思路是先用中序遍历先将二叉树的节点值输出到数组中，然后用双指针去累加数组元素及其后面的节点，用其和代替该数组元素，然后再把新的有序数组转化为BST.

后来，看了Carl的视频，发现直接在二叉树上操作也可以。

代码如下：

class Solution {
private:
    int pre = 0;//pre指针用来记录cur前一个节点的值，用于累加
    void traversal(TreeNode*cur) {
        if (cur == nullptr)return;
        traversal(cur->right);
        cur->val += pre;
        pre = cur->val;
        traversal(cur->left);
    }
public:
    TreeNode* convertBST(TreeNode*& root) {
        int pre = 0;
        traversal(root);
        return root;
    }
};

669. 修剪二叉搜索树

力扣题目链接(opens new window)

给定一个二叉搜索树，同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树，使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点，所以结果应当返回修剪好的二叉搜索树的新的根节点。

核心思想是：

如果该节点的值小于区间下限，说明该节点左孩子的值小于区间下限，右孩子的值不一定，所以把右孩子继续递归判断，同理，如果该节点的值小于区间下限，说明该节点右孩子的值大于区间上限，左孩子的值不一定，所以把左孩子继续递归判断。

 

代码如下：

class Solution {
public:
    TreeNode* trimBST(TreeNode*& root, int low, int high) {
        if (root == nullptr) return NULL;
        if (root->val < low) {
            root->right = trimBST(root->right, low, high);  //如果该节点的值小于区间下限，说明该节点左孩子的值小于区间下限，右孩子的值不一定，所以把右孩子继续递归判断
            return root->right;
        }
        if (root->val > high) {
            root->left = trimBST(root->left, low, high);  //如果该节点的值小于区间下限，说明该节点右孩子的值大于区间上限，左孩子的值不一定，所以把左孩子继续递归判断
            return root->left;
        }
        root->left = trimBST(root->left, low, high);
        root->right = trimBST(root->right, low, high);
        return root;
    }
};