代码随想录打卡第二十一天

代码随想录–二叉树部分

day 21 二叉树第八天

---

---

一、力扣669–修建二叉搜索树

代码随想录题目链接：代码随想录

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

这里不能想的太复杂，要注意是二叉搜索树，如果某个节点比low还小，那说明它左子树可以整个删了，反之亦然

那么向下搜索，如果遇到小于low的节点，就递归它的右子树，大于high就左子树

至于删除节点，只需要在return的时候注意一下不是return root，而是return递归结果

只不过这样没法释放内存，会有些浪费

代码如下：

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if(!root) return nullptr;
        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–将有序数组转换为二叉搜索树

代码随想录题目链接：代码随想录

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

核心思路就是递归构建，并且每次都对nums进行分割，分成左子树数组和右子树数组，和构建最大二叉树一个想法

代码如下：

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

三、力扣538–把二叉搜索树转换为累加树

代码随想录题目链接：代码随想录

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

简单讲就是把每个大于当前节点的值都加起来，和当前节点相加作为该节点的新值

那么只要从右叶子节点向左边遍历，每遍历一次就对sum累加一次，加到最后就结束了

这里sum定义成全局变量，记录遍历以来每个node的值之和

也就是反过来的中序递归

代码如下：

class Solution {
public:
    int sum = 0;
    void traversal(TreeNode * curr)
    {
        if(!curr) return;
        traversal(curr->right);
        curr->val += sum;
        sum = curr->val;
        traversal(curr->left);
    }
    TreeNode* convertBST(TreeNode* root) {
        traversal(root);
        return root;
    }
};

原文地址：https://blog.csdn.net/weixin_48013375/article/details/140344319

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！