题目
给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L)
。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。
思路
递归法
-
递归函数的参数和返回值:根节点和需要修剪的区间
-
终止条件:遇到空节点返回即可,因为修剪的操作不是在终止条件上进行的
-
单层递归的逻辑:如果root的元素小于low 的值, 则递归右子树,并返回右子树符合条件的头结点;反之递归左子树
java代码如下:
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在[low,high]范围内
root.left = trimBST(root.left,low,high);// root->left接入符合条件的左孩子,从而实现删除操作
root.right = trimBST(root.right,low,high);// root->right接入符合条件的右孩子,从而实现删除操作
return root;
}
}