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

iceland.JPG

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

输入格式:

输入首先在第一行给出一个正整数 N(1),为当地人口数。随后 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

首先要懂题意,五代以内不存在共同祖先的意思要弄明白,另外要避免超时,剪枝很重要。

代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <unordered_map>
using namespace std;
char fname[21],lname[21],fname1[21],lname1[21];
int n,m,no,flag;
unordered_map<string,int> pos;
bool gender[200001];//1男0女
int f[200001];
bool dfs(int a,int b,int h1,int h2) {
    if(!b || h1 >= 5 && h2 >= 5) return 1;
    if(a == b) {
        flag = 1;//直接可以确定答案
        if(h1 >= 5 && h2 >= 5) return 1;//共同祖先在五代之外
        else return 0;//共同祖先在五代之内
    }
    return dfs(a,f[b],h1,h2 + 1);
}
bool check(int a,int b) {
    flag = 0;
    int h = 1,tag = 1;
    while(a && !flag) {
        tag &= dfs(a,b,h,1);
        a = f[a];
        h ++;
    }
    return tag;
}
int main() {
    scanf("%d",&n);
    for(int i = 0;i < n;i ++) {
        scanf("%s%s",fname,lname);
        if(!pos[fname]) pos[fname] = ++ no;
        int len = strlen(lname),d = pos[fname];
        if(lname[len - 1] == 'n') {
            gender[d] = 1;
            lname[len - 4] = 0;
            if(!pos[lname]) {
                pos[lname] = ++ no;
                gender[no] = 1;
            }
            f[d] = pos[lname];
        }
        else if(lname[len - 1] == 'r') {
            gender[d] = 0;
            lname[len - 7] = 0;
            if(!pos[lname]) {
                pos[lname] = ++ no;
                gender[no] = 1;
            }
            f[d] = pos[lname];
        }
        else if(lname[len - 1] == 'm') gender[d] = 1;
        else gender[d] = 0;
    }
    scanf("%d",&m);
    for(int i = 0;i < m;i ++) {
        scanf("%s%s%s%s",fname,lname,fname1,lname);
        int a = pos[fname],b = pos[fname1];
        if(!a || !b) printf("NA\n");
        else if(gender[a] == gender[b]) printf("Whatever\n");
        else printf("%s\n",check(a,b) ? "Yes" : "No");
    }
}