学习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
$