L2-030 冰岛人

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

L2-030 冰岛人_第1张图片

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

输入格式:

输入首先在第一行给出一个正整数 N(1名 姓(带性别后缀),两个字符串均由不超过 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

#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
map fa;
map sex;
int n;
string check(string s1, string s2) {
	string s3 = s1, s4 = s2;
	set ss1, ss2;
	while (!s1.empty()) {
		ss1.insert(s1);
		s1 = fa[s1];
	}
	while (!s2.empty()) {
		ss2.insert(s2);
		s2 = fa[s2];
	}
	for (int i = 0; i < 4 && !s3.empty(); i++) {
		if (ss2.count(s3)) return "No";
		s3 = fa[s3];
	}
	for (int i = 0; i < 4 && !s4.empty(); i++) {
		if (ss1.count(s4)) return "No";
		s4 = fa[s4];
	}
	return "Yes";
}
int main() {

	cin >> n;
	string s1, s2, s3;
	for (int i = 0; i < n; i++) {
		cin >> s1 >> s2;
		if (s2.back() == 'm' || s2.back() == 'f') {
			sex[s1] = s2.back();
		}
		else if (s2.back() == 'n') {
			s2.erase(s2.size() - 4, 4);
			fa[s1] = s2;
			sex[s1] = 'm';
		}
		else {
			s2.erase(s2.size() - 7, 7);
			fa[s1] = s2;
			sex[s1] = 'f';
		}
	}
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> s1 >> s3 >> s2 >> s3;
		if (!sex.count(s1) || !sex.count(s2)) {
			printf("NA\n");
		}
		else if (sex[s1] == sex[s2]) {
			printf("Whatever\n");
		}
		else {
			cout << check(s1, s2) << endl;;
		}
	}
	return 0;
}

你可能感兴趣的:(算法题,算法,c++)