给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树不应该改变保留在树中的元素的相对结构(即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在唯一的答案。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trim-a-binary-search-tree
这道题目,一看二叉搜索树就知道可以用递归。
因为修建节点可能要删除节点,这涉及到树结构的恢复,所以返回值是树节点。
确定结束条件是节点为空,就返回空。
- 如果节点值小于可取的最小值,那么该节点的左孩子部分都要删除,右孩子可能有符合条件的节点。于是递归右孩子,将递归的返回值返回。
- 如果节点值大于可取的最大值,那么该节点的右孩子部分都要删除,递归左孩子,将递归返回值返回。
- 如果当前节点值符合取值范围,那么就递归左右孩子,分别用root.left,root.right接收递归的返回值。 然后返回当前节点。
每一层递归都会把该层符合条件的值返回给上一层,这样就可以确保返回的值都是符合取值条件的。
上面不符合取值条件的时候要递归可能大家比较难理解,是这样,每一次不符合条件,函数都会去查找它的孩子节点是否有符合条件的,如果有就返回给上一层,一直递归查找下去,如果递归到空,就代表没有符合条件的节点,就返回空,然后会一直把结果返回到已经确定的节点,成为它的左孩子或者右孩子。
总结:递归函数的作用是把每一层循环符合条件的节点返回给上一层,如果该层循环的值不符合,就会去它的孩子节点找符合的值,如果都没有符合的就返回空。
凡是涉及到节点的删除的题,都可以尝试用这种把符合条件的节点返回给上一层的做法。
源码:
class Solution {
public TreeNode trimBST(TreeNode root, int low, int high) {
if(root==null){
return null;
}
if(root.val<low){
return trimBST(root.right,low,high);
}
if(root.val>high){
return trimBST(root.left,low,high);
}
root.left=trimBST(root.left,low,high);
root.right=trimBST(root.right,low,high);
return root;
}
}