洛谷刷题C++语言 | P2814 家谱

​学习C++从娃娃抓起！记录下洛谷C++学习和备考过程中的题目，记录每一个瞬间。

附上汇总贴：洛谷刷题C++语言 | 汇总

---

【题目描述】

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

【输入】

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

【输出】

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

【输入样例】

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

【输出样例】

Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur

【代码详解】

#include <bits/stdc++.h>
using namespace std;
const int N = 50005;
int p[N];
string name[N], s;
map<string, int> mp;
int id=1, fa;
int find(int x)
{
    return p[x]==x ? x : p[x]=find(p[x]);
}
int main()
{
    for (int i=1; i<=49999; i++) {
        p[i] = i;
    }
    while (1) {
        cin >> s;
        if (s[0]=='$') break;
        string tmp = s.substr(1);
        if (s[0]=='#') {
            if (mp.count(tmp)==0) {
                mp[tmp] = id;
                name[id] = tmp;
                fa = id;
                id++;
            } else fa = find(mp[tmp]);
        } else if (s[0]=='+') {
            if (mp.count(tmp)==0) {
                mp[tmp] = id;
                name[id] = s;
                p[id] = fa;
                id++;
            } else p[mp[tmp]] = fa;  //作为孩子之前出现过
        } else if (s[0]=='?') {
            cout << tmp << " " << name[find(mp[tmp])] << endl;
        }
    }
    return 0;
}

【运行结果】

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