【Lintcode】1311. Lowest Common Ancestor of a Binary Search Tree

题目地址：

https://www.lintcode.com/problem/lowest-common-ancestor-of-a-binary-search-tree/description

给定一个二叉搜索树（假定无重复元素），再给定树中的两个节点，求其最近公共祖先。

不妨设那两个节点的值分别是$p$和$q$并且$p<q$，我们证明最近公共祖先$x$一定可以这样找到：首先令$x$为树根，若$x$满足$p\le x\le q$，那么$x$就是最近公共祖先；若$x<p$则向右子树找，若$x>q$则向左子树找；不停重复上面操作直到$x$满足$p\le x\le q$为止。
证明如下：

首先，若$x$满足$p\le x\le q$，那么由二叉搜索树的性质，要么$x$本身就是$p$或者$q$，此时显然成立；要么$p$在$x$左子树中，$q$在$x$右子树中，此时显然$x$的左子树任意一个节点都不可能是$q$的祖先，同样，$x$的右子树任意一个节点都不可能是$p$的祖先，也就是他们的公共祖先是唯一的，显然结论也成立；

若$x<p$，那么说明$p$和$q$都在$x$的右子树中，所以$x$的右子树根一定也是$p$和$q$的公共祖先，并且深度比$x$深，所以可以直接向$x$右子树走一步，继续寻找；若$x>q$推理类似。

由于每次走一步时到达的节点都仍然是公共祖先，而树的总节点数是有限的，所以向下走的步数必然也是有限的，也就是说必然走到某处就会停止，而停下来的地方就是答案。所以算法正确。

代码如下：

public class Solution {
    /**
     * @param root: root of the tree
     * @param p: the node p
     * @param q: the node q
     * @return: find the LCA of p and q
     */
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // write your code here
        if (p.val > q.val) {
            TreeNode tmp = p;
            p = q;
            q = tmp;
        }
        
        while (root != null) {
            if (root.val < p.val) {
                root = root.right;
            } else if (root.val > q.val) {
                root = root.left;
            } else {
                return root;
            }
        }
        
        return root;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    TreeNode(int x) {
        val = x;
    }
}

时间复杂度$O(h)$，空间$O(1)$。