leetcode刷题（41）——236. 二叉树的最近公共祖先

一、题目

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

例如，给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]

说明:

所有节点的值都是唯一的。
p、q 为不同节点且均存在于给定的二叉树中。

二、思路

思路就是不断地去找 p 和 q 的位置，到底在哪个子树中，找的过程中，可能会返回null、p、q、普通 Node。

• 返回 p、q 就说明 p、q 在当前节点的子树中（或当前节点就是）；
• 返回普通 Node 说明找到了 p、q 的最近公共祖先；
• 返回 null 说明 p、q 不在该节点的子树。

寻找的过程中，首先是判断当前 root。如果满足 root 为 null 或为 p 或为 q ，则直接返回。之后会递归判断 root 的 left，进行剪枝判断是否需要递归 root.right，如果需要则会递归判断 root.right。

剪枝函数（递归结束条件）：

if(root==null||root==p||root == q)
	return root;

如果 root 为 p 或 q 就可以直接返回，不需要判断 root 的左右子树，因为即使另一个节点在 root 的左子树，root 也是他们的最近公共节点。

三、代码

class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        if(root == null || root == p || root == q)
            return root;
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if(left == null)
            return right;
        else if(right == null)
            return left;
        else
            return root;
    }
}