【Lintcode】474. Lowest Common Ancestor II

题目地址：

https://www.lintcode.com/problem/lowest-common-ancestor-ii/description

给定一个二叉树，求两个节点$A$和$B$的最近公共祖先。题目保证两个节点都属于这棵二叉树，同时每个节点有一个指向其parent的指针。

由树的性质知，唯一存在从树根到$A$和$B$的路径。设树根为$r$，$r$到$A$的路径为$r=A_0\to A_1\to A_2\to ...\to A$，设$r$到$B$的路径为$r=B_0\to B_1\to B_2\to ...\to B$，那么由于$r=A_0=B_0$，显然存在某个$k$使得$A_{k-1}=B_{k-1}$但$\forall l\ge k$，有$A_l\ne B_l$（或者一条路径完全包含在另一条路径里）。理由是，因为树没有环，所以一旦两条路径开始分叉，就永远不会再相遇了。即$A_{k-1}$就是$A$和$B$的最近公共祖先。所以我们的算法是，先求出两个节点的深度，让深的那个向上走，一直到两者一样深，再一起向上走，直到相等为止。代码如下：

public class Solution {
    /*
     * @param root: The root of the tree
     * @param A: node in the tree
     * @param B: node in the tree
     * @return: The lowest common ancestor of A and B
     */
    public ParentTreeNode lowestCommonAncestorII(ParentTreeNode root, ParentTreeNode A, ParentTreeNode B) {
        // write your code here
        int dA = depth(root, A);
        int dB = depth(root, B);
        
        if (dA > dB) {
            while (dA != dB) {
                A = A.parent;
                dA--;
            }
        } else {
            while (dA != dB) {
                B = B.parent;
                dB--;
            }
        }
        
        while (A != B) {
            A = A.parent;
            B = B.parent;
        }
        
        return A;
    }
    
    private int depth(ParentTreeNode root, ParentTreeNode node) {
        int res = 0;
        while (root != node) {
            node = node.parent;
            res++;
        }
        
        return res;
    }
}

class ParentTreeNode {
    public ParentTreeNode parent, left, right;
}

时间复杂度$O(h)$。