给出一个二叉树，输入两个树节点，求它们的最低公共祖先。

一个树节点的祖先节点包括它本身。

注意：

  • 输入的二叉树不为空；
  • 输入的两个节点一定不为空，且是二叉树中的节点；

样例

 1
 2
 3
 4
 5
 6
 7
 8
 9
10

二叉树[8, 12, 2, null, null, 6, 4, null, null, null, null]如下图所示：
    8
   / \
  12  2
     / \
    6   4

1. 如果输入的树节点为2和12，则输出的最低公共祖先为树节点8。

2. 如果输入的树节点为2和6，则输出的最低公共祖先为树节点2。

代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

/**
 * 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) {
        return f(root,p,q);
    }
    #define Node TreeNode
    
    Node* f(Node*root,Node*p,Node* q) {
        if(!root) return NULL;
        if(p==root || q == root) return root;
        auto left = f(root->left,p,q);
        auto right = f(root->right,p,q);
        if(left && right)
            return root;
        return left? left:right;
    }
};