题目来源:
https://www.luogu.org/problemnew/show/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个字符,而且首字母大写,且没有任意两个人的名字相同。最多可能有1000组父子关系,总人数最多可能达到50000人,家谱中的记载不超过30代。
解题思路:
这题是裸的并查集(压缩路径),就是把字符串转化数字就行,可是我一开始还没看出来。。。。我用两个map来搞字符串,其他就是并查集的模板了。。
代码:
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <map>
using namespace std;
map<string,int>m1;map<int,string>m2;
int father[50005];
void init()
{
for(int i=1;i<=50000;i++)
father[i]=i;
}
int fi(int x)
{
if(x==father[x])return x;
else return father[x]=fi(father[x]);
}
bool same(int x,int y)
{
if(fi(x)==fi(y))return 1;
return 0;
}
void Union(int x,int y)
{
int u=fi(x),v=fi(y);
if(u==v)return ;
father[x]=y;
}
int main()
{
init();string a,b,d;int gs=0;
char c;
while((c=getchar())!=EOF)
{
if(c=='$')break;
if(c=='#'){
a.clear();
cin>>a;
if(!m1[a])m1[a]=++gs,m2[m1[a]]=a;
}
if(c=='+'){
b.clear();
cin>>b;
if(!m1[b])m1[b]=++gs,m2[m1[b]]=b;
Union(m1[b],m1[a]);
}
if(c=='?'){
cin>>d;
cout<<d<<" "<<m2[fi(m1[d])]<<endl;
}
}
return 0;
}