洛谷P2814 家谱(并查集) 题解

题目来源:

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;
}

猜你喜欢

转载自blog.csdn.net/qq_40400202/article/details/81102090