好的,今天给大家带来依旧是 水题 愉悦心情的题啊。
家谱
Description
现代的人对于本家族血统越来越感兴趣,现在给出充足的父子关系,请你编写程序找到某个人的最早的祖先。
Input
输入文件由多行组成,首先是一系列有关父子关系的描述,其中每一组父子关系由二行组成,用#name的形式描写一组父子关系中的父亲的名字,用+name的形式描写一组父子关系中的儿子的名字;接下来用?name的形式表示要求该人的最早的祖先;最后用单独的一个$表示文件结束。规定每个人的名字都有且只有6个字符,而且首字母大写,且没有任意两个人的名字相同。最多可能有1000组父子关系,总人数最多可能达到50000人,家谱中的记载不超过30代。
Output
按照输入文件的要求顺序,求出每一个要找祖先的人的祖先,格式:本人的名字+一个空格+祖先的名字+回车。
Samples
Input
#George
+Rodney
#Arthur
+Gareth
+Walter
#Gareth
+Edward
?Edward
?Walter
?Rodney
?Arthur
$
Output
Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur
思路:用个并查集然后路径压缩一下就行,在来个map映射一波。
Code:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5e5;
map<int,string> x;
map<string,int> q;
string s,temp1,temp2,t3;
int cnt=0;
int p[N];
int find(int n) {
if(n==p[n]) return n;
return p[n] = find(p[n]);
}
int main() {
while(cin>>s) {
if(s[0]=='$') break;
if(s[0]=='#') {
temp1 = s.substr(1);//把名字提出来
if(q[temp1]>0){
continue; //被标记过的话直接跳过即可
}
q[temp1] = ++cnt;
p[cnt] = cnt;
x[cnt] = temp1;
}
if(s[0]=='+') {
temp2 = s.substr(1);
q[temp2] = ++cnt;
p[cnt] = cnt;
x[cnt] = temp2;
int d1 = find(p[cnt]);
int d2 = find(p[q[temp1]]);
p[d1] = p[d2];
}
if(s[0]=='?') {
t3 = s.substr(1);
cout<<t3<<" "<<x[p[q[t3]]];
putchar('\n');
}
}
return 0;
}
ovo