洛谷 P2814 家谱

家谱

题目背景

现代的人对于本家族血统越来越感兴趣。

题目描述

给出充足的父子关系，请你编写程序找到某个人的最早的祖先。

输入格式

输入由多行组成，首先是一系列有关父子关系的描述，其中每一组父子关系中父亲只有一行，儿子可能有若干行，用 #name 的形式描写一组父子关系中的父亲的名字，用 +name 的形式描写一组父子关系中的儿子的名字；接下来用 ?name 的形式表示要求该人的最早的祖先；最后用单独的一个 $ 表示文件结束。

输出格式

按照输入文件的要求顺序，求出每一个要找祖先的人的祖先，格式为：本人的名字 $+$ 一个空格 $+$ 祖先的名字 $+$ 回车。

样例 #1

样例输入 #1

#George
+Rodney
#Arthur
+Gareth
+Walter
#Gareth
+Edward
?Edward
?Walter
?Rodney
?Arthur
$

样例输出 #1

Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur

提示

规定每个人的名字都有且只有 $6$ 个字符，而且首字母大写，且没有任意两个人的名字相同。最多可能有 $10^3$ 组父子关系，总人数最多可能达到 $5\times 10^4$ 人，家谱中的记载不超过 $30$ 代。

解题思路

可以利用map 来保存每个儿子的父亲是谁，然后递归查找祖先即可，类似于并查集的思想。

#include<iostream>
#include<string>
#include<map>
using namespace std;

map<string, string>mp;

string find(string s)
{
    if(mp[s]=="") return s;
    return find(mp[s]);
}
int main()
{
    string s, fa;
    cin >> s;
    while(s[0]!='$')
    {
        if(s[0]=='#')
        {
            fa = s.substr(1);
        }
        else if(s[0]=='+')
        {
            mp[s.substr(1)] = fa;
        }
        else if(s[0]=='?')
        {
            //cout<<find(s.substr(1)) << endl;
            cout << s.substr(1) << " " << find(s.substr(1)) << endl;
        }
        cin>>s;
    }    
    return 0;
}