【2020年天梯赛—校内选拔赛】7-12 冰岛人

题目

2018年世界杯，冰岛队因1:1平了强大的阿根廷队而一战成名。好事者发现冰岛人的名字后面似乎都有个“松”（son），于是有网友科普如下：
冰岛人沿用的是维京人古老的父系姓制，孩子的姓等于父亲的名加后缀，如果是儿子就加 sson，女儿则加 sdottir。因为冰岛人口较少，为避免近亲繁衍，本地人交往前先用个 App 查一下两人祖宗若干代有无联系。本题就请你实现这个 App 的功能。

输入格式

输出格式

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

若两人为异性，且五代以内无公共祖先，则输出 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<bits/stdc++.h>
using namespace std;
struct dot{
	int sex;
	string father;
};
map<string,dot>mp;
string fname,lname;
int judge(string s1,string s2){
	int i=1,j;
	for( string A=s1; !A.empty(); i++,A=mp[A].father ){
		j=1;
		for( string B=s2; !B.empty(); j++,B=mp[B].father ){
			if(i>=5&&j>=5) return 1;
			if(A==B&&(i<5||j<5)) return 0;
		}
	}
	return 1;
}
int main(){
	int n,m;scanf("%d",&n);
	for( int i=0; i<n; i++ ){
		cin>>fname>>lname;
		if(lname.back()=='m') mp[fname].sex=1;
		else if(lname.back()=='f') mp[fname].sex=-1;
		else if(lname.back()=='n'){
			string buf=lname.substr(0,lname.length()-4);
			mp[fname]={1,buf};
		}
		else if(lname.back()=='r'){
			string buf=lname.substr(0,lname.length()-7);
			mp[fname]={-1,buf};
		}
	}
	scanf("%d",&m);
	for( int i=0; i<m; i++ ){
		string f1,l1,f2,l2;
		cin>>f1>>l1>>f2>>l2;
		if(mp.find(f1)==mp.end()||mp.find(f2)==mp.end()) printf("NA\n"); 
		else if(mp[f1].sex==mp[f2].sex) printf("Whatever\n");
		else printf("%s\n",judge(f1,f2)?"Yes":"No");
	}
	return 0;
}