【刷题】牛客CD176-最近公共祖先节点。

题目描述
给定一棵二叉树以及这棵树上的两个节点 o1 和 o2，请找到 o1 和 o2 的最近公共祖先节点。
输入描述:
第一行输入两个整数 n 和 root，n 表示二叉树的总节点个数，root 表示二叉树的根节点。
以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)
最后一行为节点 o1 和 o2。
输出描述:
输出一个整数表示答案。
输入
8 1
1 2 3
2 4 5
4 0 0
5 0 0
3 6 7
6 0 0
7 8 0
8 0 0
4 5
输出
2

#include<bits/stdc++.h>
using namespace std;
struct TreeNode{
    int value;
    TreeNode*left,*right;
    TreeNode(int value_):value(value_),left(NULL),right(NULL){}
};
map<int,TreeNode*>mp;

TreeNode* nearestCommonRoot(TreeNode*root,TreeNode*node1,TreeNode*node2){
    //后序遍历
    //没发现最近公共祖先时，返回的是距离o1或o2最远的祖先
    //发现后，返回最近公共祖先
    if(root==NULL||root==node1||root==node2)
        return root;
    TreeNode*left = nearestCommonRoot(root->left, node1, node2);
    TreeNode*right = nearestCommonRoot(root->right, node1, node2);
    if(left==NULL&&right==NULL)
        return NULL;
    else if(left==NULL||right==NULL)
        if(left==NULL)
            return right;
        else 
            return left;
    else return root;

}

TreeNode*createNode(int value){
    TreeNode* cur;
    if(mp.count(value)==0){
        cur = new TreeNode(value);
        mp[value] = cur;
    }else{
        cur = mp[value];
    }
    return  cur;
}
int main(){
    int n,r;
    cin>>n>>r;
    TreeNode* root = new TreeNode(r);
    mp[r] = root;
    for(int i=0;i<n;i++){
        int fa,lch,rch;
        scanf("%d %d %d",&fa,&lch,&rch);
        TreeNode* cur = createNode(fa);
        if(lch){
            cur->left = createNode(lch);
        }else{
            cur->left =NULL;
        }
        if(rch){
            cur->right = createNode(rch);
        }else{
            cur->right=NULL;
        }
    }
    int o1,o2;
    cin>>o1>>o2;
    TreeNode*res = nearestCommonRoot(root,mp[o1], mp[o2]);
    cout<<res->value<<endl;
}