669. 修剪二叉搜索树
思路:给定一个二叉搜索树,同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。举个例子:
Input:
3
/ \
0 4
\
2
/
1
L = 1
R = 3
Output:
3
/
2
/
1
先看一下我的错误案例,最初的想法是,只要不满足区间的直接设置为null即可:
public TreeNode trimBST(TreeNode root, int L, int R) {
if(root == null) return root;
if(root.val<L || root.val>R) return null;
TreeNode left = trimBST(root.left,L,R);
TreeNode right = trimBST(root.right,L,R);
root.left = left;
root.right = right;
return root;
}
显然这是错误的,就例如上面我举得例子,这个代码运行后只能输出3,这是因为不满足条件的子树就直接被裁掉了,然而这样是不对的,子树的树根不满足并不能代表他的子节点也不满足,所以导致多剪了不该剪的枝。->
那么应该怎么剪枝呢?我们要充分利用二叉搜索树的特性,如果某个子节点的值小于左边界,那么就向它的右边递归,因为他的右边一定是更大的。同理,如果某个子节点的值大于右边界,那么就向它的左边递归,因为他的左边一定是更小的。正确代码如下:
public TreeNode trimBST(TreeNode root, int L, int R) {
if(root == null) return root;
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;
}
接下来一题来帮助我们更好的理解二叉搜索树的性质
转换成更大二叉树
思路:转换规则如下:
Input: The root of a Binary Search Tree like this:
5
/ \
2 13
Output: The root of a Greater Tree like this:
18
/ \
20 13
可以看出要用递归的方法,我们直到二叉搜索树的中序遍历结果就是其已排好序的结果,所以我们只需要进行一遍中序遍历,不断从后累加即可。
又由于我们要从后累加(从大累加),所以中序遍历的时候要从右到左输出,即先遍历右节点再遍历左节点,完整代码如下:
public int sum = 0;
public TreeNode convertBST(TreeNode root) {
traver(root);
return root;
}
public void traver(TreeNode node) {
if (node == null) return;
traver(node.right);
sum += node.val;
node.val = sum;
traver(node.left);
}
235. 二叉搜索树的最近公共祖先
思路:给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”例如:
_______6______
/ \
___2__ ___8__
/ \ / \
0 4 7 9
/ \
3 5
2和8的最近公共祖先是6,2和4的最近公共祖先是2
由于题设是二叉搜索树,所以我们可以利用二叉搜索树的特点,如果p、q的值都小于root,说明p q 肯定在root的左子树中;如果p q都大于root,说明肯定在root的右子树中,如果一个在左一个在右则说明此时的root记为对应的最近公共祖先。->
那么我们如何表示p q是在左还是右呢? 首先我们从根节点开始比起,将