【PTA-天梯賽訓練】願天下有情人都是失散多年的兄妹

呵呵。大家都知道五服以內不得通婚，即兩個人最近的共同祖先如果在五代以內（即本人、父母、祖父母、曾祖父母、高祖父母）則不可通婚。本題就請你幫助一對有情人判斷一下，他們究竟是否可以成婚？

輸入格式：

輸入第一行給出一個正整數N（2 ≤ N ≤），隨後N行，每行按以下格式給出一個人的信息：

本人ID 性別 父親ID 母親ID

其中ID是5位數字，每人不同；性別M代表男性、F代表女性。如果某人的父親或母親已經不可考，則相應的ID位置上標記為-1。

接下來給出一個正整數K，隨後K行，每行給出一對有情人的ID，其間以空格分隔。

註意：題目保證兩個人是同輩，每人只有一個性別，並且血緣關係網中沒有亂倫或隔輩成婚的情況。

輸出格式：

對每一對有情人，判斷他們的關係是否可以通婚：如果兩人是同性，輸出Never Mind；如果是異性並且關係出了五服，輸出Yes；如果異性關係未出五服，輸出No。

輸入樣例：

24
00001 M 01111 -1
00002 F 02222 03333
00003 M 02222 03333
00004 F 04444 03333
00005 M 04444 05555
00006 F 04444 05555
00007 F 06666 07777
00008 M 06666 07777
00009 M 00001 00002
00010 M 00003 00006
00011 F 00005 00007
00012 F 00008 08888
00013 F 00009 00011
00014 M 00010 09999
00015 M 00010 09999
00016 M 10000 00012
00017 F -1 00012
00018 F 11000 00013
00019 F 11100 00018
00020 F 00015 11110
00021 M 11100 00020
00022 M 00016 -1
00023 M 10012 00017
00024 M 00022 10013
9
00021 00024
00019 00024
00011 00012
00022 00018
00001 00004
00013 00016
00017 00015
00019 00021
00010 00011

輸出樣例：

Never Mind
Yes
Never Mind
No
Yes
No
Yes
No
No

思路：兩個dfs函數，先搜第一個人的五代親屬並標記，再找另一個的五代親屬是否與其重覆。

#include<bits/stdc++.h>
using namespace std;
int vis[100005],f;
struct peo
{
    int fu,mu;
    char sex;
    peo():fu(-1),mu(-1){}    //結構體初始化，一定要先將父母的值都賦值為-1，不然會遞歸到id為0的地方，發生錯誤！！
}a[100005];
int dfs(int x,int y)         //先找出第一個人的五代親屬 
{
    if(y>5)
        return 0;
    vis[x]=1;                //親屬位置標記為1 
    if(a[x].fu!=-1)
        dfs(a[x].fu,y+1);
    if(a[x].mu!=-1)
        dfs(a[x].mu,y+1);
} 
int dfss(int x,int y)        //再找出第二個人的五代親屬 
{
    if(y>5)
        return 0;
    if(vis[x]==0)
    {
        if(a[x].fu!=-1)
            dfss(a[x].fu,y+1);
        if(a[x].mu!=-1)
            dfss(a[x].mu,y+1);
    }
    else 
    {
        f=1;                 //如果訪問過，標記為1，跳出
        return 0;
    }
}
int main()
{
    int i,x,y,k,id,fu,mu,n;
    char sex;
    cin>>n;
    for(i=1;i<=n;i++)
    {
        scanf("%d %c %d %d",&id,&sex,&fu,&mu);
        a[id].fu=fu;
        a[id].mu=mu;
        a[id].sex=sex;
        a[fu].sex='M';        //一定要給父母的性別賦值！！！！
        a[mu].sex='F';
    }
    cin>>k;
    while(k--)
    {
        scanf("%d%d",&x,&y);
        if(a[x].sex==a[y].sex)
        {
            printf("Never Mind\n");
        }
        else 
        {
            memset(vis,0,sizeof(vis));
            f=0;
            dfs(x,1);
            dfss(y,1);
            
            if(f==1)
            printf("No\n");
            else printf("Yes\n");
        }
    }
    return 0;
}