算法笔记|Day18二叉树VIII
☆☆☆☆☆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,这样在每次递归后才能保留更改后的值;若采用局部变量,每次递归调用时独立创建和销毁,它们之间不会相互影响。