夸父追日：第六章 二叉树part08

今日收获：修剪二叉搜索树，将有序数组转换为二叉搜索树，把二叉搜索树转换为累加树

1. 修剪二叉搜索树

题目链接：669. - 力扣（LeetCode）

思路：根据root的值判断，如果大于high则保留左子树，如果小于low则保留右子树，在区间中则修剪左右子树

方法：

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root==null){
            return null;
        }

        // root大于high，留左子树
        if (root.val>high){
            return trimBST(root.left,low,high);
        }

        // root小于low，留右子树
        if (root.val<low){
            return trimBST(root.right,low,high);
        }

        // root在区间中
        root.left=trimBST(root.left,low,high);
        root.right=trimBST(root.right,low,high);

        return root;
    }
}

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

题目链接：108. - 力扣（LeetCode）

思路：寻找数组中间的节点作为根节点，思路类似于构造二叉树

方法：

class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return sortedArrayToBST(nums,0,nums.length);
    }

    public TreeNode sortedArrayToBST(int[] nums,int begin,int end){
        // 寻找中间节点作为根节点，数组左闭右开
        if (begin>=end){
            return null;
        }


        int index=(end-begin)/2+begin;
        TreeNode root=new TreeNode(nums[index]);
        root.left=sortedArrayToBST(nums,begin,index);
        root.right=sortedArrayToBST(nums,index+1,end);
        return root;
    }
}

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

题目链接：538. - 力扣（LeetCode）

思路：反中序 “右中左” 遍历，定义全局指针记录当前遍历节点的上一个节点

方法：

class Solution {
    TreeNode pre;

    public TreeNode convertBST(TreeNode root) {
        // 反中序遍历
        convert(root);
        return root;
    }

    public void convert(TreeNode node){
        if (node==null){
            return;
        }

        convert(node.right);

        if (pre!=null){
            node.val+=pre.val;
        }
        pre=node;

        convert(node.left);
    }
}