冰岛人沿用的是维京人古老的父系姓制,孩子的姓等于父亲的名加后缀,如果是儿子就加 sson,女儿则加 sdottir。因为冰岛人口较少,为避免近亲繁衍,本地人交往前先用个 App 查一下两人祖宗若干代有无联系。本题就请你实现这个 App 的功能。
输入格式:
输入首先在第一行给出一个正整数 N(1<N≤105),为当地人口数。随后 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
注意:
- “五代之内” 指的是四代(自己 -> 爸爸 -> 爸爸的爸爸 -> 爸爸的爸爸的爸爸)
- 查询并没有保证两个人是同辈。
- 共同祖先(有的话)不能是任何一方的五代之内,比如一个是 8 代,另一个是 3 代,这就是不行的。
- 题目可能会问你,某人能不能和他的祖先结婚。
#include <bits/stdc++.h>
using namespace std;
struct Node
{
bool sex;
string father;
};
int n, m;
string a, b, c, d;
map<string, Node> mp; // 记录性别和父亲
set<string> s1, s2; // 把第一个人的五代内的亲属和自己装入s1, 五代之外的亲属装 s2
int main()
{
ios::sync_with_stdio(false);
cin >> n;
while (n--)
{
cin >> a >> b;
if (b.back() == 'm')
mp[a].sex = 1;
else if (b.back() == 'f')
mp[a].sex = 0;
else if (b.back() == 'n')
mp[a] = {1, b.substr(0, b.size() - 4)};
else
mp[a] = {0, b.substr(0, b.size() - 7)};
}
cin >> m;
while (m--)
{
cin >> a >> b >> c >> d;
if (mp.find(a) == mp.end() || mp.find(c) == mp.end())
cout << "NA" << endl;
else if (mp[a].sex == mp[c].sex)
cout << "Whatever" << endl;
else
{
bool flag = 0;
s1.clear(), s2.clear();
string name = a;
for (int i = 0; i < 4 && !name.empty(); ++i)
{
s1.insert(name);
name = mp[name].father;
}
while (!name.empty())
{
s2.insert(name);
name = mp[name].father;
}
name = c;
for (int i = 0; i < 4 && !name.empty(); ++i)
{
if (s1.find(name) != s1.end() || s2.find(name) != s2.end())
{
flag = 1;
break;
}
name = mp[name].father;
}
while (!flag && !name.empty())
{
if (s1.find(name) != s1.end())
{
flag = 1;
break;
}
name = mp[name].father;
}
if (flag)
cout << "No" << endl;
else
cout << "Yes" << endl;
}
}
return 0;
}