题目描述
给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
例如,给定如下二叉搜索树: root = [6,2,8,0,4,7,9,null,null,3,5]
样例描述
思路
递归 + 分情况讨论
时间复杂度 O(h)h为树的高度
- 考虑两个结点三种位置情况,两个分别在两边,都在左子树,或者都在右子树
- 由于BST且树的结点值互不相同,可以预先保证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) {
if (p.val > q.val) {
TreeNode t = p;
p = q;
q = t;
}
//如果一个在根结点左边,一个在根结点右边,根结点就是LSA
if (p.val <= root.val && q.val >= root.val) {
return root;
}
//都在左子树/右子树,就直接递归
//大的那个都小于根,肯定就都在左边
if (q.val < root.val) {
return lowestCommonAncestor(root.left, p, q);
}
return lowestCommonAncestor(root.right, p ,q);
}
}