C语言：L2-016 愿天下有情人都是失散多年的兄妹 (25 分) — 并查集+dfs

一、题目

呵呵。大家都知道五服以内不得通婚，即两个人最近的共同祖先如果在五代以内（即本人、父母、祖父母、曾祖父母、高祖父母）则不可通婚。本题就请你帮助一对有情人判断一下，他们究竟是否可以成婚？

输入格式：

输入第一行给出一个正整数 N（2 ≤ N ≤10⁴ ），随后 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

二、方法1

1、思路

看到这种家庭关系第一时间想到的就是并查集，然后再根据题意，五代之内不能通婚，很明显想到了深度优先搜索。

深度优先搜索：

1. 双方同时开始找祖先，如果五代以内不相交，则可以通婚；反之，则不行。
2. 一个人找到五代以内所有祖先，放入一个数组，另一个人再开始一一对比，如果都不一样，则可以通婚；反之，则不行。

易错点提示：

• 在定义孩子的父母时记得将父母初始化为-1；
• 并查集数组也要进行初始化；
• 并查集的实现中，find 函数如果用 while 循环来实现，最后会超时，所以可以选用递归（只要内存限制不大，递归可以省很多时间）；

2、代码

#include<stdio.h>
#define max 100000

typedef struct Node
{
    int father;                             // 父亲
    int mather;                             // 母亲
    char sex;                               // 性别
}node;
node a[max];

int n, k;
int i, j;                                   // 遍历
int id, father, mather;                     // 孩子，父，母
char sex;                                   // 性别
int x, y;                                   // 需要判断的两个人

int pre[max];                               // 并查集数组
void init();
int find(int x);
void Union(int a, int b);

// 深度优先搜索
int dfs(int x, int y, int age);

int main()
{
    init();
    scanf("%d", &n);

    for (i = 0; i < n; i++)
    {
        scanf("%d %c %d %d", &id, &sex, &father, &mather);
        // 存值
        a[id].father = father;
        a[id].sex = sex;
        a[id].mather = mather;

        // 将孩子与父母联系起来
        Union(id, father);
        Union(id, mather);

        // 父母的性别
        if (father != -1)
            a[father].sex = 'M';
        if (mather != -1)
            a[mather].sex = 'F';
    }

    scanf("%d", &k);

    for (i = 0; i < k; i++)
    {
        scanf("%d %d", &x, &y);
        // 先判断性别
        if (a[x].sex == a[y].sex)
        {
            printf("Never Mind\n");
        }
        else
        {
            // 不同祖先
            if (find(x) != find(y))
                printf("Yes\n");
            // 相同祖先
            else if (dfs(x, y, 1))
                printf("Yes\n");
            else
                printf("No\n");
        }
    }
    return 0;
}

void init()
{
    for (int i = 0; i < max; i++)
    {
        pre[i] = i;
        a[i].father = -1;
        a[i].mather = -1;
    }
}


int find(int x)
{
    if(x == pre[x])
        return x;
    return pre[x] = find(pre[x]);
}

void Union(int a, int b)
{
    if (b == -1)
        return;
    int f_a = find(a);
    int f_b = find(b);
    if (f_a != f_b)
    {
        pre[f_a] = f_b;
    }
}

int dfs(int x, int y, int age)
{
    if (!(x != -1 && y != -1))
        return 1;

    if (age > 5)
        return 1;
    if (x == y)
    {
        return 0;
    }
    else
        return dfs(a[x].father, a[y].father, age + 1) && dfs(a[x].father, a[y].mather, age + 1) && dfs(a[x].mather, a[y].father, age + 1) && dfs(a[x].mather, a[y].mather, age + 1);
}