PTA L2-030 冰岛人 (并查集)

                                                 L2-030 冰岛人

2018年世界杯，冰岛队因1:1平了强大的阿根廷队而一战成名。好事者发现冰岛人的名字后面似乎都有个“松”（son），于是有网友科普如下：

冰岛人沿用的是维京人古老的父系姓制，孩子的姓等于父亲的名加后缀，如果是儿子就加 sson，女儿则加 sdottir。因为冰岛人口较少，为避免近亲繁衍，本地人交往前先用个 App 查一下两人祖宗若干代有无联系。本题就请你实现这个 App 的功能。

输入格式：

输入首先在第一行给出一个正整数 N（1<N≤10​5​​），为当地人口数。随后 N 行，每行给出一个人名，格式为：名 姓（带性别后缀），两个字符串均由不超过 20 个小写的英文字母组成。维京人后裔是可以通过姓的后缀判断其性别的，其他人则是在姓的后面加 m 表示男性、f 表示女性。题目保证给出的每个维京家族的起源人都是男性。

随后一行给出正整数 M，为查询数量。随后 M 行，每行给出一对人名，格式为：名1 姓1 名2 姓2。注意：这里的姓是不带后缀的。四个字符串均由不超过 20 个小写的英文字母组成。

题目保证不存在两个人是同名的。

输出格式：

对每一个查询，根据结果在一行内显示以下信息：

• 若两人为异性，且五代以内无公共祖先，则输出 Yes；
• 若两人为异性，但五代以内（不包括第五代）有公共祖先，则输出 No；
• 若两人为同性，则输出 Whatever；
• 若有一人不在名单内，则输出 NA。

所谓“五代以内无公共祖先”是指两人的公共祖先（如果存在的话）必须比任何一方的曾祖父辈分高。

输入样例：

15
chris smithm
adam smithm
bob adamsson
jack chrissson
bill chrissson
mike jacksson
steve billsson
tim mikesson
april mikesdottir
eric stevesson
tracy timsdottir
james ericsson
patrick jacksson
robin patricksson
will robinsson
6
tracy tim james eric
will robin tracy tim
april mike steve bill
bob adam eric steve
tracy tim tracy tim
x man april mikes

输出样例：

Yes
No
No
Whatever
Whatever
NA

 AC Code:

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <queue>
#include <climits>
#include <set>
#include <stack>
#include <string>
#include <map>
#include <vector>
using namespace std;
typedef long long ll;
static const int MAX_N = 1e5 + 5;
struct Per {
	bool sex;
	int id;
};
map<string, Per>mps1;
int pre[MAX_N];
string s1, s2, s3, s4;
void init() {
	for (int i = 1; i < MAX_N; ++i)pre[i] = i;
}
int Find(int x) {
	int rt = pre[x];
	if (pre[x] != x) rt = Find(pre[x]);
	return rt;
}
bool judge(int x, int y) {	
	/*
		两人存在公共祖先的情况下
		判断这个共同祖先是否在五代以内
	*/int ret = 1;
	while (x != pre[x] && y != pre[y]) {
		x = pre[x];
		y = pre[y];
		++ret;
		if (ret == 5) return true;	//不在五代以内
		if (x == y) return false;	//五代以内存在共同祖先
	}
	return false;	
/*
	能够跳出while循环
	说明某个人无五代祖先
	但是已经判断出两人存在公共祖先
	说明两人五代以内存在公共祖先,返回false
*/
}
int main() {
	ios::sync_with_stdio(false);
	/*freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);*/
	init();
	int n;
	cin >> n;
	int cnt = 0;
	vector<vector<string> >in(n);
	for (int i = 0; i < n; ++i) {
		cin >> s1 >> s2;
		mps1[s1].id = ++cnt;
		in[i].push_back(s1);
		in[i].push_back(s2);
	}
	for (int i = 0; i < n; ++i) {
		s1 = in[i][0], s2 = in[i][1];
		if (s2.length() > 4 && s2.substr(s2.length() - 4, 4) == "sson") {
			pre[mps1[s1].id] = mps1[s2.substr(0, s2.length() - 4)].id;
			mps1[s1].sex = true;
		}
		else if (s2.length() > 7 && s2.substr(s2.length() - 7, 7) == "sdottir") {
			pre[mps1[s1].id] = mps1[s2.substr(0, s2.length() - 7)].id;
			mps1[s1].sex = false;
		}
		else {
			if (s2.back() == 'm')	mps1[s1].sex = true;
			else mps1[s1].sex = false;
		}
	}
	int m;
	/*for (map<string, Per>::iterator it = mps1.begin(); it != mps1.end(); ++it) {
		cout << it->first << " " << it->second.id << " " << pre[it->second.id] << endl;
	}*/
	cin >> m;
	while (m--) {
		cin >> s1 >> s2 >> s3 >> s4;
		if (!mps1[s1].id || !mps1[s3].id) printf("NA\n");
		else if (mps1[s1].sex == mps1[s3].sex) printf("Whatever\n");
		else if (Find(mps1[s1].id) != Find(mps1[s3].id)) printf("Yes\n");
		else if (judge(mps1[s1].id, mps1[s3].id)) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}

另一种解法：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <queue>
#include <climits>
#include <set>
#include <stack>
#include <string>
#include <map>
#include <vector>
using namespace std;
typedef long long ll;
static const int MAX_N = 1e5 + 5;
struct Per {
	bool sex;
	int id;
};
map<string, Per>mps1;
int pre[MAX_N];
int n;
string s1, s2, s3, s4;
void init() {
	for (int i = 1; i < MAX_N; ++i) pre[i] = i;
}
bool Not_Find(int tx, int ty) {
/*
	判断五代之内是否有公共祖先：
	A的第十代和B的第二代祖先相同，也代表A、B五代以内有公共祖先
	所以是指任意一方五代以内均无对方公共祖先
*/	
	if (tx == ty) return false;
	int x = tx, y = ty;
	set<int>s;
	vector<int>lev1(n + 1, 0);
	vector<int>lev2(n + 1, 0);
	s.insert(x);
	while (x != pre[x]) {
		lev1[pre[x]] = lev1[x] + 1;
		x = pre[x];
		s.insert(x);
	}
	if (s.count(y)) return false;
	while (y != pre[y]) {
		lev2[pre[y]] = lev2[y] + 1;
		y = pre[y];
		if (s.count(y)) {
			if (lev1[y] >= 4 && lev2[y] >= 4) return true;
			else return false;
		}
	}
	return true;
}
int main() {
	ios::sync_with_stdio(false);
	/*freopen("input.txt", "r", stdin);
	freopen("output.txt", "w", stdout);*/
	init();
	cin >> n;
	int cnt = 0;
	vector<vector<string> >in(n);
	/*
		这里需要把所有的输入存储，第6个测试点含有这样的数据：
		孩子的名、姓已经输入了，但父亲的名、姓还没有输入
		也就是后者的数据在前面的后面输入
		这样的话，孩子去找父亲的编号就是0了
	*/
	for (int i = 0; i < n; ++i) {
		cin >> s1 >> s2;
		mps1[s1].id = ++cnt;	//先编号，存储输入input
		in[i].push_back(s1);
		in[i].push_back(s2);
	}
	for (int i = 0; i < n; ++i) {
		s1 = in[i][0], s2 = in[i][1];
		if (s2.length() > 4 && s2.substr(s2.length() - 4, 4) == "sson") {
			pre[mps1[s1].id] = mps1[s2.substr(0, s2.length() - 4)].id;
			mps1[s1].sex = true;
		}
		else if (s2.length() > 7 && s2.substr(s2.length() - 7, 7) == "sdottir") {
			pre[mps1[s1].id] = mps1[s2.substr(0, s2.length() - 7)].id;
			mps1[s1].sex = false;
		}
		else {
			if (s2[s2.length() - 1] == 'm')	mps1[s1].sex = true;
			else mps1[s1].sex = false;
		}
	}
	int m;
	/*for (map<string, Per>::iterator it = mps1.begin(); it != mps1.end(); ++it) {
		cout << it->first << " " << it->second.id << " " << pre[it->second.id] << endl;
	}*/
	cin >> m;
	while (m--) {
		cin >> s1 >> s2 >> s3 >> s4;
		if (mps1.find(s1) == mps1.end() || mps1.find(s3) == mps1.end()) printf("NA\n");
		else if (mps1[s1].sex == mps1[s3].sex) printf("Whatever\n");
		else if (Not_Find(mps1[s1].id, mps1[s3].id)) printf("Yes\n");
		else printf("No\n");
	}
	return 0;
}