2018年世界杯,冰岛队因1:1平了强大的阿根廷队而一战成名。好事者发现冰岛人的名字后面似乎都有个“松”(son),于是有网友科普如下:
iceland.JPG
冰岛人沿用的是维京人古老的父系姓制,孩子的姓等于父亲的名加后缀,如果是儿子就加 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
注意:
1:(2测试点错误原因)不是维京人的没有爸爸妈妈(我们不能用维京人的方式判断他爸妈是谁)。
2:(3,6测试点错误原因)题目强调:“五代以内无公共祖先”包含两个意思,以下两种情况都yes。
2-1:有公共祖先,且公共祖先是属于某一方(或双方)的5代以上。
2-2:无公共祖先。
解答:
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
#include <vector>
#define foreach(a,b,c) for((a)=(b); (a)<(c); (a)++)
#define MAX 100002
using namespace std;
int sexs[MAX];
map<string,int> name_index;
int parent[MAX];
struct Names{
string name;
string surname;
}names[MAX];
void formatSurname(char *surname,int *sex,int length);
bool isRootNotInFive(int pos_a,int pos_b);
int main(){
char a_name[21],a_surname[21],b_name[21],b_surname[21];
int i,k,n,sex;
foreach(i,0,MAX) {parent[i]=-1;sexs[i]=0;} // 0男
scanf("%d",&n);
map<string,int> menu;
foreach(i,0,n){
scanf("%s %s",a_name,a_surname);
menu.insert(pair<string,int>{a_name,1});
names[i].name = a_name;
names[i].surname = a_surname;
formatSurname(a_surname, &sex, strlen(a_surname));
name_index[a_name] = i+1;
sexs[name_index[a_name]] = sex;
}
foreach(i,0,n){
if(names[i].surname[names[i].surname.length()-1] == 'n' || names[i].surname[names[i].surname.length()-1] == 'r'){
char *temp = (char *)names[i].surname.c_str();
formatSurname(temp, &sex, names[i].surname.length());
parent[name_index[names[i].name]] = name_index[temp];
}
}
scanf("%d",&k);
foreach(i,0,k){
scanf("%s %s %s %s",a_name,a_surname,b_name,b_surname);
bool isFindA = menu.find(a_name) != menu.end();
bool isFindB = menu.find(b_name) != menu.end();
if(isFindA && isFindB){
if(sexs[name_index[a_name]] == sexs[name_index[b_name]]){
printf("Whatever\n");
}else if(isRootNotInFive(name_index[a_name],name_index[b_name])){
printf("Yes\n");
}else{
printf("No\n");
}
}else{
printf("NA\n");
}
}
return 0;
}
bool isRootNotInFive(int pos_a,int pos_b){
int cnt=1,meet[MAX]={0},a_parent[MAX]={0},b_parent[MAX]={0};
while(pos_a != -1 && pos_a != 0){
meet[pos_a] = 1;
if(cnt < 5) a_parent[pos_a] = 1;
if(pos_a == pos_b) return false;
pos_a = parent[pos_a];
cnt ++;
}
cnt = 1;
while(pos_b != -1 && pos_a != 0){
if(cnt < 5) b_parent[pos_b] = 1;
if(meet[pos_b] == 1){
if(a_parent[pos_b] == 1 || b_parent[pos_b] == 1) return false;
}
pos_b = parent[pos_b];
cnt ++;
}
return true;
}
void formatSurname(char *surname,int *sex,int length){
int flag = 0;
switch(surname[length-1]){
case 'm':
flag = 1;
/* no break */
case 'n':
if(flag == 1) surname[length-1]='\0';
else (surname[length-4]='\0',surname[length-3]='\0',surname[length-2]='\0',surname[length-1]='\0');
*sex = 0;
break;
case 'f':
flag = 1;
/* no break */
case 'r':
if(flag == 1) surname[length-1]='\0';
else (surname[length-7]='\0',surname[length-6]='\0',surname[length-5]='\0',
surname[length-4]='\0',surname[length-3]='\0',surname[length-2]='\0',surname[length-1]='\0');
*sex = 1;
}
}