hiho刷题日记——第十三天最近公共祖先·一

题目

已知一棵族谱树。查找两个人的最近公共祖先。

输入

每个测试点（输入文件）有且仅有一组测试数据。
每组测试数据的第1行为一个整数N，意义如前文所述。
每组测试数据的第2~N+1行，每行分别描述一对父子关系，其中第i+1行为两个由大小写字母组成的字符串Father_i, Son_i，分别表示父亲的名字和儿子的名字。
每组测试数据的第N+2行为一个整数M，表示小Hi总共询问的次数。
每组测试数据的第N+3~N+M+2行，每行分别描述一个询问，其中第N+i+2行为两个由大小写字母组成的字符串Name1_i, Name2_i，分别表示小Hi询问中的两个名字。
对于100%的数据，满足N<=10^2，M<=10^2, 且数据中所有涉及的人物中不存在两个名字相同的人（即姓名唯一的确定了一个人）。

输出

对于每组测试数据，对于每个小Hi的询问，输出一行，表示查询的结果：如果根据已知信息，可以判定询问中的两个人存在共同的祖先，则输出他们的所有共同祖先中辈分最低的一个人的名字，否则输出-1。

思路

首先建立一棵族谱树。对于输入的父子关系做记录：father[son_name]=father_name;
然后开始找最近公共祖先。对于查询两个人（name1，name2）用一下方式查找：
1.将name1及其所有的祖先做标记mark。
2.查询name2的祖先，找到的第一个被做了标记的祖先便是他们最近的公共祖先。若找到底都没找到，则name1和name2没有公共祖先。

代码

#include<cstdio> 
#include<iostream>
#include<string>
#include<vector>
#include<map>

using namespace std;
int N;

map<string,string> fa;
map<string,int> mask;

int main()
{

    string fn,sn;
    scanf("%d",&N);
    while(N--)
    {
        cin>>fn>>sn;
        fa[sn]=fn;
        mask[sn]=0;
        mask[fn]=0;
    } 

    scanf("%d",&N);
    while(N--)
    {
        for(map<string,int>::iterator ite = mask.begin(); ite != mask.end(); ite++) 
        {   
            ite->second=0;
        }
        cin>>fn>>sn;

        while(fa.count(fn))
        {
            mask[fn]=1;
            fn=fa[fn];
        }
        mask[fn]=1;
        while(fa.count(sn) && !mask[sn])
        {
            sn=fa[sn];
        }
        if(mask[sn]) cout<<sn<<endl;
        else cout<<-1<<endl; 
    }

    return 0;
}