Nameless Site

But one day, you will stand before its decrepit gate,without really knowing why.

0%

二叉搜索树最近的公共祖先

来自Leetcode第235题二叉搜索树最近的公共祖先

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”

例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]

img

示例 1:

1
2
3
输入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8
输出: 6
解释: 节点 2 和节点 8 的最近公共祖先是 6。

递归

  1. 从根节点开始遍历树
  2. 如果节点 pp 和节点 qq 都在右子树上,那么以右孩子为根节点继续 1 的操作
  3. 如果节点 pp 和节点 qq 都在左子树上,那么以左孩子为根节点继续 1 的操作
  4. 如果条件 2 和条件 3 都不成立,这就意味着我们已经找到节 pp 和节点 qq 的 LCA 了
1
2
3
4
5
6
7
8
9
10
11
12
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
int pval = p.val;
int qval = q.val;
int rval = root.val;

if(pval > rval && qval > rval)
return lowestCommonAncestor(root.right,p,q);
else if(pval < rval && qval < rval)
return lowestCommonAncestor(root.left,p,q);
else
return root;
}

迭代

思路同上

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
int pval = p.val;
int qval = q.val;
int rval = root.val;

while(root != null) {
rval = root.val;
if (pval > rval && qval > rval)
//return lowestCommonAncestor(root.right, p, q);
root = root.right;
else if (pval < rval && qval < rval)
//return lowestCommonAncestor(root.left, p, q);
root = root.left;
else
return root;
}
return null;
}