修剪二叉搜索树
题目
修剪二叉搜索树(力扣:669)
给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。
分析
二叉搜索树:它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉搜索树。
我们根据它的特点:
- root为null时,直接返回null。
- 节点小于最小范围,则需要修剪,返回修剪后的右子树。
- 节点大于最大范围,则需要修剪,返回修剪后的左子树。
- 最后修整左、右子节点,返回节点。
代码实现
/**
* 669. 修剪二叉搜索树
* @param root
* @param L
* @param R
* @return
*/
public TreeNode trimBST(TreeNode root, int L, int R) {
if(root == null){
return null;
}
//节点小于最小范围,则需要修剪,返回修剪后的右子树。
if(root.val < L){
return trimBST(root.right, L, R);
}
//节点大于最大范围,则需要修剪,返回修剪后的左子树。
if(root.val > R){
return trimBST(root.left, L, R);
}
//修整左、右子节点
root.left = trimBST(root.left, L, R);
root.right = trimBST(root.right, L, R);
return root;
}