代码随想录算法训练营第二十天|669. 修剪二叉搜索树、108.将有序数组转换为二叉搜索树、538.把二叉搜索树转换为累加树、总结篇

=669. 修剪二叉搜索树 

这道题目比较难，比 添加增加和删除节点难的多，建议先看视频理解。

题目链接/文章讲解： 代码随想录

视频讲解： 你修剪的方式不对，我来给你纠正一下！| LeetCode：669. 修剪二叉搜索树_哔哩哔哩_bilibili

 这里我们为什么需要返回值呢？

因为是要遍历整棵树，做修改，其实不需要返回值也可以，我们也可以完成修剪（其实就是从二叉树中移除节点）的操作。

但是有返回值，更方便，可以通过递归函数的返回值来移除节点。

修剪的操作并不是在终止条件上进行的，所以就是遇到空节点返回就可以了。

二叉树移除节点的过程，是通过返回删除节点的右孩子或左孩子给要删除节点的父节点，从而达到要移除节点的效果。

此外，要删除节点的左\右子树可能还符合区间，要进一步向下一层去递归

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if(root == null) return root;
        
        // 如果root（当前节点）的元素小于low的数值，那么应该递归右子树，并返回右子树符合条件的头结点。
        if(root.val < low){
            TreeNode right = trimBST(root.right, low, high);// 寻找符合区间[low, high]的节点
            return right;
        }

        // 如果root(当前节点)的元素大于high的，那么应该递归左子树，并返回左子树符合条件的头结点。
        if(root.val > high){
            TreeNode left = trimBST(root.left, low, high);// 寻找符合区间[low, high]的节点
            return left;
        }

        // root在[low,high]范围内
        //将下一层处理完左子树的结果赋给root->left，处理完右子树的结果赋给root->right。
        root.left = trimBST(root.left, low, high);// root->left接入符合条件的左孩子
        root.right = trimBST(root.right, low, high);// root->right接入符合条件的右孩子

        return root;//root在上面就处理了
    }
}

---

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

本题就简单一些，可以尝试先自己做做。

题目链接/文章讲解：代码随想录

视频讲解：构造平衡二叉搜索树！| LeetCode：108.将有序数组转换为二叉搜索树_哔哩哔哩_bilibili

 本质就是寻找分割点，分割点作为当前节点，然后递归左区间和右区间。有序数组构造二叉搜索树，寻找分割点就比较容易了。分割点就是数组中间位置的节点。

如果数组长度为偶数，中间节点有两个，取哪一个？

取哪一个都可以，只不过构成了不同的平衡二叉搜索树。

切割区间时，注意区间的定义。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return sortedArrayToBSTImp(nums, 0, nums.length);
    }

    // 左闭右开区间[left, right)
    public TreeNode sortedArrayToBSTImp(int[] nums, int left, int right) {
        if(left >= right) return null;
        if(right - left == 1) return new TreeNode(nums[left]);

        // 如果数组长度为偶数，中间位置有两个元素，取靠左边的。
        int middle = left + ((right - left) / 2);//防止数据越界，如left和right都是最大int，这么操作就越界了
        TreeNode root = new TreeNode(nums[middle]);

        root.left = sortedArrayToBSTImp(nums, left, middle);
        root.right = sortedArrayToBSTImp(nums, middle + 1, right);

        return root;
    }
}

---

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

本题也不难，在 求二叉搜索树的最小绝对差 和 众数 那两道题目 都讲过了 双指针法，思路是一样的。

代码随想录题目链接/文章讲解：代码随想录

视频讲解：普大喜奔！二叉树章节已全部更完啦！| LeetCode：538.把二叉搜索树转换为累加树_哔哩哔哩_bilibili

 本题依然需要一个pre指针记录当前遍历节点cur的前一个节点，这样才方便做累加。

不需要递归函数的返回值做什么操作了，要遍历整棵树。

同时需要定义一个全局变量pre，用来保存cur节点的前一个节点的数值，定义为int型就可以了。

注意要右中左来遍历二叉树， 中节点的处理逻辑就是让cur的数值加上前一个节点的数值。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int pre = 0;// 记录前一个节点的数值

    public TreeNode convertBST(TreeNode root) {
        traversal(root);
        return root;
    }

    // 按右中左顺序遍历，累加即可
    public void traversal(TreeNode cur){// 右中左遍历
        if(cur == null) return;

        convertBST(cur.right);// 右

        cur.val += pre;// 中
        pre = cur.val;

        convertBST(cur.left);// 左
    }
}

---

总结篇  

好了，二叉树大家就这样刷完了，做一个总结吧

代码随想录文章讲解：代码随想录

 在每一道二叉树的题目中，都要使用递归三部曲来分析题目，以后看到二叉树，看到递归，要想：返回值、参数是什么？终止条件是什么？单层逻辑是什么？

• 涉及到二叉树的构造，无论普通二叉树还是二叉搜索树一定前序，都是先构造中节点。

• 求普通二叉树的属性，一般是后序，一般要通过递归函数的返回值做计算。

• 求二叉搜索树的属性，一定是中序了，要不白瞎了有序性了。

注意在普通二叉树的属性中，我用的是一般为后序，例如单纯求深度就用前序，二叉树：找所有路径 (opens new window)也用了前序，这是为了方便让父节点指向子节点。

所以求普通二叉树的属性还是要具体问题具体分析。