题目描述:
给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树 不应该 改变保留在树中的元素的相对结构 (即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在 唯一的答案 。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。
来源:力扣(LeetCode)
思路:使用递归容易实现
一个重点:
一开始所有的BST题目,使用递归,自己首先就相使用无返回值的方法;
但是做了许多之后,所有的递归都是用了有返回值TreeNode的方法,理解对返回的节点的接收就好!
1)划分了区间:BST的题容易想到使用 中序排好序的特性进行处理
2)根节点不在区间中,说明修剪之后的结果在左右其中一个;将子节点作为新的root进行递归就好
3)根节点在区间中,则需要保留,对左右子树分别处理之后返回根节点就好!
代码:
1)递归实现
class Solution {
//做BST的题时,进行递归自己总是先想到记录当前节点,再分别传入左右进行递归!
//这样的实现很麻烦!
//只用将递归函数的返回值设置为 节点,就可以将左右子树进行处理!
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root == null) return null;//走到了叶子节点
//此时root不为null
//此时修剪后的区间在右子树中
if(root.val < low) return trimBST(root.right,low,high);
//修剪后的区间在左子树中
else if(root.val > high) return trimBST(root.left,low,high);
//此时根节点在 区间中
else{
root.left = trimBST(root.left,low,high);
root.right = trimBST(root.right,low,high);
//左右子树修改之后,返回根节点
return root;
}
}
}