题目描述:
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
我们来看一下二叉搜索树的特征:对于一个结点而言,它的左孩子的值一定比它小,右孩子的值一定比它大。
故根据这个特点我们可以将题目整理成三种情况:
- 当给定的p q 结点的 val 都比 root 的值小,则说明应该在根结点的左子树中找最近公共祖先。(因为根结点的左孩子结点,不一定是 p q 的最近公共祖先,故仍需向下遍历)
- 当给定的p q 结点的 val 都比 root 的值大,则说明应该在根结点的右子树中找最近公共祖先。(右子树也许向下遍历)
- 当 p q 结点分在当下根结点的左右两边时,即可说明该结点就是最近公共祖先。
代码如下👇
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
//根结点的值
int parentVal = root.val;
//p结点的值
int pVal = p.val;
//q结点的值
int qVal = q.val;
if (pVal > parentVal && qVal > parentVal) {
//pq结点都在根结点的右子树上,则从根结点的右孩子往下遍历
return lowestCommonAncestor(root.right, p, q);
} else if (pVal < parentVal && qVal < parentVal) {
//pq结点都在根结点的左子树上,则从根结点的左孩子往下遍历
return lowestCommonAncestor(root.left, p, q);
} else {
//如果他们两个没在一边,则说明找到了最近结点值
return root;
}
}
}