[Lintcode] Lowest Common Ancestor

Lowest Common Ancestor

Given the root and two nodes in a Binary Tree. Find the lowest common ancestor(LCA) of the two nodes.

The lowest common ancestor is the node with largest depth which is the ancestor of both nodes.

Example

For the following binary tree:

  4
 / \
3   7
   / \
  5   6

LCA(3, 5) = 4

LCA(5, 6) = 7

LCA(6, 7) = 7

SOLUTION:

这个题首先要看面试官有没有给parent指针，如果有的话当然比较好做，我们这里先来说一下不给parent指针的做法。

思路1：

首先我们先来看A，B两个点位置有几种情况：

1，A，B在一个分支上，可以都在左子树里面，也可以都在右子树里面，这时候的LCA就是他们其中最靠上的点自己。

2，A，B不在一个分支上，那么LCA就是他们分叉位置的那个root点。

3，当然还有一种可能是A，B不在树上面，当然返回null。

代码实现过程中还是用到了Divide & Conquer，因为我们要看A，B点是否在一个分支上。所以实现的时候要先Divide，然后再做中止条件的处理，递归的返回条件就是找到了A或者B就返回这个root点。

思路2：

换一种想法，更直接的体现本题的意义。结果分为三种情况（去除特殊情况外）：

1，只在左子树发现LCA，返回左子树这个LCA；

2，只在右子树发现LCA，返回右子树这个LCA；

3，左子树，右子树都没有发现LCA（左子树，右子树同时有值返回，那么两个点肯定都不是LCA），返回root点；

具体实现看代码：

public class Solution {
    /**
     * @param root: The root of the binary search tree.
     * @param A and B: two nodes in a Binary.
     * @return: Return the least common ancestor(LCA) of the two nodes.
     */
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode A, TreeNode B) {
        if (root == null || root == A || root == B){
            return root;
        }
        TreeNode left = lowestCommonAncestor(root.left, A, B);
        TreeNode right = lowestCommonAncestor(root.right, A, B);
        if (left != null && right != null){
            return root;
        }
        if (left != null){
            return left;
        }
        if (right != null){
            return right;
        }
        return null;
    }
}

作者：http://www.cnblogs.com/tritritri/

转载于:https://www.cnblogs.com/tritritri/p/4858183.html