题目描述

现代的人对于本家族血统越来越感兴趣,现在给出充足的父子关系,请你编写程序找到某个人的最早的祖先。

输入

输入文件由多行组成,首先是一系列有关父子关系的描述,其中每一组父子关系由二行组成,用#name的形式描写一组父子关系中的父亲的名字,用+name的形式描写一组父子关系中的儿子的名字;接下来用?name的形式表示要求该人的最早的祖先;最后用单独的一个$表示文件结束。规定每个人的名字都有且只有6个字符,而且首字母大写,且没有任意两个人的名字相同。最多可能有1000组父子关系,总人数最多可能达到50000人,家谱中的记载不超过30代。

输出

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

样例输入 
#George
+Rodney
#Arthur
+Gareth
+Walter
#Gareth
+Edward
?Edward
?Walter
?Rodney
?Arthur
$
样例输出 
Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur

其实这题很简单,我们肯定用并查集来模拟,那我们怎么快速将人名映射出来呢,当然是用容器来存储,这里适合用的容器是map ~~(我最喜欢的容器)~~可以映射下标:map<string,string>fa第一个string类型表示的是一对关系中儿子是谁,第二个string类型表示的是一对关系中父亲是谁。但是我们需要写一个函数来跳出他最早的祖先。

inline string getfa(string x){return fa[x]==x?x:fa[x]=getfa(fa[x]);}

这是一段记忆化DFS来查找,也是并查集的核心部分。表示如果这个点的父亲是这个点,那么说明已经找到头了,就不找了。不然就找下去,并记住他。 附我的代码:

#include<bits/stdc++.h>
using namespace std;
map<string,string> fa;
string father,now,lst;
inline string getfa(string x){return fa[x]==x?x:fa[x]=getfa(fa[x]);}
int main(){
	char ch=getchar();
	while (ch!='$'){
		cin>>now;getchar();
		if (ch=='#'){
			father=now;
			if (fa[father]=="") fa[father]=father;//如果当前这个父亲没有父亲,那么我们就认为他是自己的父亲。
		}else
		if (ch=='+') fa[now]=father;else cout<<now<<" "<<getfa(now)<<endl;
		ch=getchar();
	}
	return 0;
}

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐