代码随想录算法训练营第18天 | 第六章 二叉树 part08

第六章 二叉树 - Part 08

目录

• 669. 修剪二叉搜索树
• 108. 将有序数组转换为二叉搜索树
• 538. 把二叉搜索树转换为累加树
• 总结篇

---

669. 修剪二叉搜索树

这道题目比较难，比添加、增加和删除节点难得多，建议先看视频理解。
这题虽然看起来比较难，但是实际思路并不难，一层层的往下遍历，遇见在范围内的保留，遇见不在范围内的删除掉，如果root（当前节点）的元素小于low的数值，那么应该递归右子树，并返回右子树符合条件的头结点。如果root(当前节点)的元素大于high的，那么应该递归左子树，并返回左子树符合条件的头结点。首先，检查当前节点是否为空。如果为空，直接返回 nullptr。这是递归的终止条件，表示已经处理完该分支的所有节。如果当前节点的值大于给定的 high，说明该节点及其右子树的所有节点值都大于 high。因此，整个右子树都应该被舍弃，只需要递归处理左子树，并返回修剪后的左子树。如果当前节点的值在 low 和 high 之间，说明该节点是符合条件的，应该被保留。接下来需要分别递归地修剪该节点的左子树和右子树，修剪完之后将修剪后的左右子树重新挂在该节点上，最后返回该节点。

题目链接/文章讲解：
https://programmercarl.com/0669.%E4%BF%AE%E5%89%AA%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91.html

/**
 * 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==nullptr)
        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;
    }
};

视频讲解：
https://www.bilibili.com/video/BV17P41177ud

---

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

本题就简单一些，可以尝试先自己做做。对于一个有序数组，将其转换为高度平衡的二叉搜索树，可以选择数组的中间元素作为根节点，数组左半部分构建左子树，右半部分构建右子树。这个选择保证了树的平衡性。
这题思路很简单，就是先按照顺序的vector[],然后再生成，二叉树。通过递归的方式，将中间元素两边的子数组继续进行类似处理，直到数组范围为空为止。这题确实和之前众数的题目很类似，搞懂之前构建平衡二叉树的题目，这题也不难理解。
核心思想是选择中间元素作为根节点，确保左右子树的节点数量大致相等，从而生成一棵平衡的树。

题目链接/文章讲解：
https://programmercarl.com/0108.%E5%B0%86%E6%9C%89%E5%BA%8F%E6%95%B0%E7%BB%84%E8%BD%AC%E6%8D%A2%E4%B8%BA%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91.html

视频讲解：
https://www.bilibili.com/video/BV1uR4y1X7qL

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

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

本题也不难，在 求二叉搜索树的最小绝对差 和 众数 那两道题目都讲过了双指针法，思路是一样的。
首先先看下类加树是什么，下面这副图就很明显：中序遍历倒着数，层层递加。从树中可以看出累加的顺序是右中左，所以我们需要反中序遍历这个二叉树，然后顺序累加就可以了。

我原本想着定义迭代函数，传入结点，输出值，结果发现，不需要递归函数的返回值做什么操作了，要遍历整棵树。因此说没必要返回值。同时需要定义一个全局变量pre，用来保存cur节点的前一个节点的数值，

题目链接/文章讲解：
https://programmercarl.com/0538.%E6%8A%8A%E4%BA%8C%E5%8F%89%E6%90%9C%E7%B4%A2%E6%A0%91%E8%BD%AC%E6%8D%A2%E4%B8%BA%E7%B4%AF%E5%8A%A0%E6%A0%91.html

视频讲解：
https://www.bilibili.com/video/BV1d44y1f7wP

class Solution {
public:
    int pre=0;
    void traversal(TreeNode* root){
        if (cur == NULL) return;
        traversal(cur->right);  // 右
        cur->val += pre;        // 中
        pre = cur->val;
        traversal(cur->left);   // 左
    }
    TreeNode* convertBST(TreeNode* root) {
        pre = 0;
        traversal(root);
        return root;
    }
};

另外全局变量，在调用函数的时候，一定要别忘了初始化，否则不能多次调用。。

总结篇

好了，二叉树就这样刷完了，做一个总结吧。二叉树确实挺难。学了很多，反正慢慢学吧。今天实在没有多少时间了，索性随便水一篇。整体来说，二叉树真的好难，思路还是懂了，但是代码敲得还是不算熟练。反正不断学习，慢慢进步。 加油加油加油！越学越是焦虑，还是一点点进步吧。

总结链接：
https://programmercarl.com/%E4%BA%8C%E5%8F%89%E6%A0%91%E6%80%BB%E7%BB%93%E7%AF%87.html