算法笔记|Day18二叉树VIII

☆☆☆☆☆leetcode 669. 修剪二叉搜索树

题目链接：leetcode 669. 修剪二叉搜索树

题目分析

涉及到二叉搜索树多个节点的删除，若当前节点值大于high，返回左节点（左节点比当前节点小，可能在要求区间内）；若当前节点值小于high，返回右节点（右节点比当前节点大，可能在要求区间内）。

代码

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if(root==null)
            return null;
        if(root.val>high)
            return trimBST(root.left,low,high);
        if(root.val<low)
            return trimBST(root.right,low,high);
        root.left=trimBST(root.left,low,high);
        root.right=trimBST(root.right,low,high);
        return root;
    }
}

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

题目链接：leetcode 108.将有序数组转换为二叉搜索树

题目分析

递归采用前序遍历，切割数组可以采用左闭右闭区间或左闭右开区间。为了满足平衡二叉搜索树的要求，需要将中间节点作为根节点，再依次递归左右两个子树。

代码

1.左闭右闭区间
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return traversal(nums,0,nums.length-1);
    }
    public TreeNode traversal(int nums[],int left,int right){
        if(left>right)
            return null;
        int mid=(left+right)/2;
        TreeNode root=new TreeNode(nums[mid]);
        root.left=traversal(nums,left,mid-1);
        root.right=traversal(nums,mid+1,right);
        return root;
    }
}

2.左闭右开区间
class Solution {
    public TreeNode sortedArrayToBST(int[] nums) {
        return traversal(nums,0,nums.length);
    }
    public TreeNode traversal(int nums[],int left,int right){
        if(left>=right)
            return null;
        int mid=(left+right)/2;
        TreeNode root=new TreeNode(nums[mid]);
        root.left=traversal(nums,left,mid);
        root.right=traversal(nums,mid+1,right);
        return root;
    }
}

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

题目链接：leetcode 538.把二叉搜索树转换为累加树

题目分析

二叉搜索树的中序遍历（左中右）可以得到一个递增的数组，此题需要从大到小依次求和，故采用反中序遍历（右中左），再依次相加即可。

代码

class Solution {
    int pre=0;
    
    public TreeNode convertBST(TreeNode root) {
        if(root==null)
            return null;
        convertBST(root.right);
        pre+=root.val;
        root.val=pre;
        convertBST(root.left);
        return root;
    }
}

提示：
使用全局变量pre，这样在每次递归后才能保留更改后的值；若采用局部变量，每次递归调用时独立创建和销毁，它们之间不会相互影响。