P2814 家谱

题目链接

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

题目描述

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

输入格式

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

输出格式

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

输入输出样例

输入 #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×104 人，家谱中的记载不超过 30 代。

 

 

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;

const int maxn = 5e4 + 10;
int f[maxn];

int find(int x)
{
    while(x != f[x])
        x = f[x] = find(f[x]);
    return x;
}

int main()
{
    int cnt = 1;
    string s;
    int formal;
    unordered_map<string, int>mp;   //通过姓名找编号
    unordered_map<int, string>mp2;  //通过编号找姓名

    for(int i = 1; i <= maxn; i++)  //初始化
        f[i] = i;

    while(cin>>s){
        if(s == "$")break;
        if(s[0] == '#'){        //父亲
            s = s.substr(1);
            if(mp[s] == 0){     //当前字符串第一次出现
                mp[s] = cnt;    //编个号
                mp2[cnt] = s;   //存一下该编号对应的字符串
                formal = cnt++;
            }else{
                formal = mp[s]; //存当前父亲的编号
            }
        }else if(s[0] == '+'){  //儿子
            s = s.substr(1);
            if(mp[s] == 0){     //当前字符串第一次出现
                mp[s] = cnt++;  //编个号
                mp2[cnt-1] = s; //存一下该编号对应的字符串
                f[mp[s]] = formal;//当前字符串的父节点是当前父亲的编号
            }else{
                f[mp[s]] = formal;
            }
        }else{
            s = s.substr(1);
            int current = find(mp[s]);//找祖宗
            cout<<s<<" "<<mp2[f[current]]<<endl;
        }
    }
    return 0;
}