L2-016 愿天下有情人都是失散多年的兄妹 (25 分)

题目链接

#include <bits/stdc++.h>

using namespace std;

struct node
{
    string x;
    int father_id=-1;//初始化为-1，如果不初始化的话会默认为0，但是编号也可能刚好为00000
    int mother_id=-1;
} a[100010];

bool dfs(int u,int v,int k)
{
    if(u<0||v<0||k>=5)//如果父母不可考或者出五服
        return true;
    if((a[u].father_id>=0&&a[u].father_id==a[v].father_id)||(a[u].mother_id>=0&&a[u].mother_id==a[v].mother_id))//父亲或者母亲为同一个人
        return false;
    else
        return dfs(a[u].father_id,a[v].father_id,k+1)&&//递归找亲戚
        dfs(a[u].mother_id,a[v].mother_id,k+1)&&
        dfs(a[u].father_id,a[v].mother_id,k+1)&&
        dfs(a[u].mother_id,a[v].father_id,k+1);
}

int main()
{
    int n;
    cin>>n;
    for(int i=0; i<n; i++)
    {
        int t;
        cin>>t;
        cin>>a[t].x>>a[t].father_id>>a[t].mother_id;
        if(a[t].father_id!=-1)//父母的性别也要记录否则会部分正确
            a[a[t].father_id].x="M";
        if(a[t].mother_id!=-1)
            a[a[t].mother_id].x="F";
    }
    int m;
    cin>>m;
    while(m--)
    {
        int l,r;
        cin>>l>>r;
        if(a[l].x==a[r].x)
            printf("Never Mind\n");
        else
        {
            if(dfs(l,r,1))
                printf("Yes\n");
            else
                printf("No\n");
        }
    }
    return 0;
}

上面要记录父母的性别很重要，打个比方，a的父亲是b，c的父亲是d，如果样例只给出了a和c，我们通过a和c知道b是a的父亲，d是c的父亲，但是b和d并没有通过样例直接给出，如果这时候题目询问b和d是否能结婚，如果我们不记录父母的性别那么我们就无法判断b和d能否结婚，如果我们记录了父母的的性别，那我们就可以很清楚的判断不行，因为他们是同性，这就是为什么要记录父母性别的原因。