Day23 LeetCode669. 修剪二叉搜索树 108. 将有序数组转换为二叉搜索树 538. 把二叉搜索树转换为累加树

最新推荐文章于 2024-10-13 15:10:00 发布

wangjiale0912

最新推荐文章于 2024-10-13 15:10:00 发布

阅读量63

点赞数

分类专栏：算法文章标签： leetcode 算法

本文链接：https://blog.csdn.net/Kimpool/article/details/130413110

版权

算法专栏收录该内容

34 篇文章 0 订阅

订阅专栏

题目：669. 修剪二叉搜索树 - 力扣（LeetCode）

思路：

1.递归函数返回值和参数：参数就是当前节点和最小边界low和最大边界high。返回值是修剪过后的新的二叉搜索树的根节点。

2.终止条件：遇到空节点直接返回NULL。

3.单层递归逻辑：(1)当前节点的值小于low，不能直接返回NULL，因为其右子树上的节点的值可能在[low,high]区间内，所以还应该往其右子树上继续递归寻找，返回值应该是右子树修剪过后新的根节点，不能直接return root->right，而是return 递归函数(root->right)；(2)当前节点的值大于high，不能直接返回NULL，因为其左子树上的节点的值可能在[low,high]区间内，所以还应该往其左子树上继续递归寻找，返回值应该是左子树修剪过后新的根节点，不能直接return root->left，而是return 递归函数(root->left)。

代码：

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(root == NULL) return NULL;
        if(root->val < low){
            TreeNode* right = trimBST(root->right, low, high);
            return right;//返回右子树修剪过后新的根节点
        }
        if(root->val > high){
            TreeNode* left = trimBST(root->left, low, high);
            return left;//返回左子树修剪过后新的根节点
        }
        root->left = trimBST(root->left, low, high);
        root->right = trimBST(root->right, low, high);
        return root;
    }
};

题目：108. 将有序数组转换为二叉搜索树 - 力扣（LeetCode）

思路：

因为要转换成一颗平衡二叉搜索树，所以需要以中序遍历的方式从数组中间元素开始递归地构建二叉树。

1.递归函数参数和返回值：先找到数组的中间元素构建根节点，然后递归地构建左子树和右子树，所以参数是有序数组和数组的左下标left和右下标right。返回值是构建好的平衡二叉树搜索树的根节点。

2.终止条件：当数组区间left>right时，递归结束返回NULL。

3.单层递归逻辑：首先取数组中间元素构建根节点，接着划分区间，root的左孩子接住下一层左区间的构造节点，右孩子接住下一层右区间构造的节点，最后返回root节点。

代码：

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

题目：538. 把二叉搜索树转换为累加树 - 力扣（LeetCode）

思路：

把二叉搜索树转换为累加树，相当于一个有序数组，然后从后往前累加得到一个有序数组。树中可以看出累加的顺序是右中左，所以我们需要反中序遍历这个二叉树，然后顺序累加就可以了。在遍历的过程中，需要用到双指针，一个pre指针记录当前遍历节点cur的前一个节点，这样才方便做累加。

代码：

class Solution {
public:
    TreeNode* pre = NULL;//pre指针记录前一个节点
    void traversal(TreeNode* cur){
        if(cur == NULL) return;
        traversal(cur->right);
        if(pre != NULL){
            cur->val = cur->val + pre->val;//当前节点和前一个节点值进行累加
        }
        pre = cur;//移动pre指针，指向当前节点
        traversal(cur->left);
    }
    TreeNode* convertBST(TreeNode* root) {
        traversal(root);
        return root;
    }
};

二叉搜索树总结：