题目背景
现代的人对于本家族血统越来越感兴趣。
题目描述
给出充足的父子关系,请你编写程序找到某个人的最早的祖先。
输入格式
输入由多行组成,首先是一系列有关父子关系的描述,其中每一组父子关系中父亲只有一行,儿子可能有若干行,用 #name 的形式描写一组父子关系中的父亲的名字,用 +name 的形式描写一组父子关系中的儿子的名字;接下来用 ?name 的形式表示要求该人的最早的祖先;最后用单独的一个 $ 表示文件结束。
输出格式
按照输入文件的要求顺序,求出每一个要找祖先的人的祖先,格式为:本人的名字 +++ 一个空格 +++ 祖先的名字 +++ 回车。
思路:
这道题第一眼看上去常规思路就是建一个树,但发现并查集的方式做也非常的简洁,于是就结合map集合,用并查集合并几个独立的团体,寻找任意一个成员的祖宗只要输出a[str]即可得出答案。
代码如下:
#include <iostream>
#include <string>
#include <map>
using namespace std;
map <string,string> a;
string abc(string str)
{
int i;
string b="";
for(i=1;i<str.size();i++)
{
b+=str[i];
}
return b;
}
string find(string str)
{
if(a[str]==str)
return str;
else {
a[str]=find(a[str]);
return a[str];
}
}
int main()
{
string str,f,z;
while(cin>>str&&str[0]!='$')
{
if(str[0]=='#')
{
f=abc(str);
if(!a.count(abc(str)))
a[abc(str)]=abc(str);
continue;
}
if(str[0]=='+')
{
a[abc(str)]=f;
}
if(str[0]=='?')
{
cout<<abc(str)<<" "<<find(abc(str))<<endl;
}
}
return 0;
}
输入输出样例
输入 #1
#George
+Rodney
#Arthur
+Gareth
+Walter
#Gareth
+Edward
?Edward
?Walter
?Rodney
?Arthur
$
输出 #1
Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur