二叉樹的最近公共祖先

給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。

百度百科中最近公共祖先的定義為：“對於有根樹 T 的兩個結點 p、q，最近公共祖先表示為一個結點 x，滿足 x 是 p、q 的祖先且 x 的深度盡可能大（一個節點也可以是它自己的祖先）。”

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

示例 1:

輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
輸出: 3
解釋: 節點 5 和節點 1 的最近公共祖先是節點 3。
示例 2:

輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
輸出: 5
解釋: 節點 5 和節點 4 的最近公共祖先是節點 5。因為根據定義最近公共祖先節點可以為節點本身。
 

說明:

所有節點的值都是唯一的。
p、q 為不同節點且均存在於給定的二叉樹中。

code1：

左右子樹的近公共結點可以是左右子樹其中任何一結點，也可是其最近公共根節點
應該采用后序遍歷，因為只有找到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||!p||!q)
            return nullptr;
        
        TreeNode* resNode=nullptr;
        lowestCommonAncestorCore(root,p,q,resNode);
        return resNode;
    }
private:
    int lowestCommonAncestorCore(TreeNode* curNode,TreeNode* p,TreeNode* q,TreeNode* &resNode)
    {
        if(!curNode)
            return 0;
    
        int left=lowestCommonAncestorCore(curNode->left,p,q,resNode);
        int right=lowestCommonAncestorCore(curNode->right,p,q,resNode);
        int cur=(curNode==p||curNode==q)?1:0;
        if(left+right+cur>=2)
            resNode=curNode;
        return left||right||cur?1:0;
    }
};

 code2：非遞歸；

1. 首先把從root到p和q的路徑都加入到parent中
2. 然后在ancestor中存儲p到根結點的路徑
3. 從q到根結點向上找，直到找到一個結點在ancestor中，該結點就是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||!p||!q)
            return nullptr;
        
        unordered_map<TreeNode*,TreeNode*> parent;
        deque<TreeNode*> stack;
        parent[root]=nullptr;
        stack.push_back(root);

        while(!stack.empty()&&(find(stack.begin(),stack.end(),p)==stack.end()||find(stack.begin(),stack.end(),q)==stack.end()))
        {
            TreeNode* node=stack.back();
            stack.pop_back();
            if(node&&node->left)
            {
                stack.push_back(node->left);
                parent[node->left]=node;
            }
            if(node&&node->right)
            {
                stack.push_back(node->right);
                parent[node->right]=node;
            }
        }

        unordered_set<TreeNode*> ancerstor;
        while(p)
        {
            ancerstor.insert(p);
            p=parent[p];
        }
        while(ancerstor.find(q)==ancerstor.end())
            q=parent[q];

        return q;
    }
};