题目:
给你二叉搜索树的根节点 root
,同时给定最小边界low
和最大边界 high
。通过修剪二叉搜索树,使得所有节点的值在[low, high]
中。修剪树 不应该 改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在 唯一的答案 。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。
思路:与删除节点类似,使用递归法。
第一步:确定参数与返回值。参数为节点,low,high,返回值为节点
第二步:确定终止条件。当节点为空时,返回null;
第三步:确定单层递归逻辑。当节点值小于low时,递归遍历其右子树;当节点值大于high时,递归遍历其左子树。然后root.left承接其符合条件的左子树,root.right承接其符合条件的右子树。
代码:
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root==null) return null;
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左子树接入符合条件的左孩子
root.right=trimBST(root.right,low,high);//root右子树接入符合条件的右孩子
return root;
}