L2-030 冰島人 (25分)

2018年世界杯，冰島隊因1:1平了強大的阿根廷隊而一戰成名。好事者發現冰島人的名字后面似乎都有個“松”（son），於是有網友科普如下：

冰島人沿用的是維京人古老的父系姓制，孩子的姓等於父親的名加后綴，如果是兒子就加 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");
    }
}