问题描述
给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。
示例 1:
输入:
1
/ \
0 2
L = 1
R = 2
输出:
1
\
2
示例 2:
输入:
3
/ \
0 4
\
2
/
1
L = 1
R = 3
输出:
3
/
2
/
1
思路
- 二叉搜索树:中序遍历时,根节点为中间值,根节点的值大于左子树的值,根节点的值小于右子树的值;
- 给定一个范围,就好比一个窗口,窗口内是有效数据,序列根节点就是一个中间标识,看根节点在窗口的位置;若根节点在窗口的右侧,则R < root.val,则右子树都剪掉,直接递归左子树返回即可;若根节点在窗口的左侧,则root.val < L,则左子树不是有效数据,剪掉,递归右子树返回即可;若在中间,则要递归更新左右子树,最终返回根节点。
实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode trimBST(TreeNode root, int L, int R){
if(root==null){
return null;
}
if(root.val>R){
return trimBST(root.left,L,R);//要返回,情况中的一种,不return,返回最先压入栈的元素
}
if(root.val<L){
return trimBST(root.right,L,R);
}
root.left = trimBST(root.left,L,R);//更新左子树
root.right = trimBST(root.right,L,R);
return root;
}
}
小结
刚开始的想法是,中序遍历,去除范围内的数据,重新递归构建二叉搜索树,但这种耗时运算比较多,不是好的方法;
现在的方法只递归修剪即可,与之前相比,少了重新遍历,创建数组的过程。