LeetCode刷题笔记 236 / 剑指Offer 68

题目：二叉树的最近公共祖先

答案：

参考链接：JAVA DFS+剪枝

递归

我们使用DFS搜索每一个节点的左右子树：
1、若子树上存在p和q的公共节点，返回此公共节点
2、若不存在公共节点，但是存在p或q任意一个节点，返回此节点
3、若不存在公共、p、q节点，则返回null。

那么，有以下几个结论：
1、若当前节点为null、p、q之一，直接返回当前节点
2、若左子树上存在公共节点（返回值非p、q），则函数返回值为左子树返回值，不需再遍历右子树（剪枝）
3、若左子树、右子树返回值均为非null，则肯定为一个p，一个q，则公共节点为当前节点，返回当前结点。
4、如果左右子树其中一个不为空，则返回非空子树节点。

*针对p是q的子节点这种特殊情况，上述方案依然可行（但就没有办法剪枝了，可以考虑针对此情况加一个标记，不再遍历右子树）。

举例：二叉树如下， p为9，q为11.

        1
       / \
     2     3
    / \   / \
  4   5   6   7
 / \  /\  /\  /\
8  9 10 11      13

以8为根的子树返回null，以9为根的子树返回9，以4为根的子树返回9；以10为根的子树返回null，以11为根的子树返回11，以5为根的子树返回11；以2为根的子树返回2（因为左右子树返回值都不为空，所以2为公共节点）以1为根的节点返回2（因为2不等于9也不等于11，所以2是已经求得的公共节点，就无需再遍历3及以下的右子树了）

代码：

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if (root == null || root == p || root == q) {          //情况1
            return root;
        }
        TreeNode left = lowestCommonAncestor(root.left, p, q); 
        if (left != null && left != q && left != p)            //情况2
            return left;
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if (left != null && right != null) {                   //情况3
            return root;
        }
        return left!=null ? left : right;                      //情况4
    }
}