【Leetcode】236. Lowest Common Ancestor of a Binary Tree

题目地址：

https://leetcode.com/problems/lowest-common-ancestor-of-a-binary-tree/

寻找一棵二叉树某两个节点$p$和$q$的最近公共祖先。

先证：$w$是$p$与$q$的最近公共祖先，当且仅当$w=p$且$q$是$p$的后代（或者反之），或者$p$与$q$分别在$w$的左右子树之中。

先证充分性：若$p$与$q$分别位于$w$的左右子树中，那么很显然$w$是公共祖先，并且$w$的左右子树根都不是公共祖先，这说明了$w$的“最近”性质。证毕。

再证必要性：反证法。不妨设$p$与$q$都位于$w$的左子树中，那么$w$的左子树根是更近的公共祖先，与$w$的“最近”性质矛盾。证毕。

所以只需要找到一个$p$和$q$分别位于它左右子树的那个节点就行了。为此我们需要在root的左右两个子树中搜索$p$和$q$，如果一边搜索出$p$另一边搜索出$q$返回即可。代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

class Solution {
 public:
  TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
    if (!root || root == p || root == q) return root;
    auto l = lowestCommonAncestor(root->left, p, q),
         r = lowestCommonAncestor(root->right, p, q);
    if (l && r) return root;
    return l ? l : r;
  }
};

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

算法正确性证明：

需要证明这个函数，若以root为树根的树中存在$p$与$q$，则会返回其最近公共祖先，若只存在$p$或$q$，则对应地返回$p$或$q$，否则会返回null。
如果root是$p$或者$q$显然正确。如果树的节点数为$0$，$1$，$2$或$3$时显然正确；假设树的节点数是$k$的时候也正确，如果树的节点数是$k+1$，那么其左右子树节点数必然小于等于$k$，据数学归纳法，若left与right里分别有$p$和$q$，那说明$p$和$q$在root的两端，那就返回root。否则按照数学归纳法，就返回了正确答案。证毕。