P2184 家谱

家谱

题目背景

现代的人对于本家族血统越来越感兴趣。

题目描述

给出充足的父子关系，请你编写程序找到某个人的最早的祖先。

输入格式

输入由多行组成，首先是一系列有关父子关系的描述，其中每一组父子关系中父亲只有一行，儿子可能有若干行，用 #name 的形式描写一组父子关系中的父亲的名字，用 +name 的形式描写一组父子关系中的儿子的名字；接下来用 ?name 的形式表示要求该人的最早的祖先；最后用单独的一个 $ 表示文件结束。

输出格式

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

样例 #1

样例输入 #1

#George
+Rodney
#Arthur
+Gareth
+Walter
#Gareth
+Edward
?Edward
?Walter
?Rodney
?Arthur
$

样例输出 #1

Edward Arthur
Walter Arthur
Rodney George
Arthur Arthur

提示

规定每个人的名字都有且只有 $6$ 个字符，而且首字母大写，且没有任意两个人的名字相同。最多可能有 $10^3$ 组父子关系，总人数最多可能达到 $5 \times 10^4$ 人，家谱中的记载不超过 $30$ 代。

---

算法1

(并查集 + map)

1.并查集思想

路径压缩，直接指向祖宗节点

string find(string x)
{
    if(x != p[x]) p[x] = find(p[x]);
    return p[x];
}

2.用map存储父节点 map<string,string> p;

3.当前读入的是父节点，那么用以个变量存储当前父节点，并且当且仅当父节点为根节点时，p[s] = s;

C++ 代码

#include <bits/stdc++.h>
using namespace std;

map<string,string> p;
char ch;
string s,sp;

string find(string x){
	if( x != p[x]) p[x] = find(p[x]);
	return p[x];
}
int main(){
	cin >> ch;
	while(ch !='$'){
		cin >> s; // 读入一个字符串
		if(ch == '#'){
			sp = s;
			if(p[s] == ""){ //当前父节点为根节点时 
				p[s] = s;
			}
		} 
		else if(ch == '+'){
			p[s] = sp;
		} 
		else{
			cout << s << " " << find(s) <<endl;
		}
		cin >> ch;
	} 
	return 0;
}

---