天梯赛 愿天下有情人都是失散多年的兄妹（Python）

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

输入格式：

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

代码长度限制

16 KB

时间限制

200 ms

内存限制

64 MB

我的思路是：

        先将所有人的父母id以字典的形式存起来，字典的键是当前人的id，再创建一个存性别的字典sexs，存所有人的性别；通过while循环每次向上找一代，从有情人当代向上取五代祖先，分别放到两个数组里，最后比较两个数组内是否有相同的元素（-1除外）。

注意（重点！！！）：

        我也在网上搜了大量的文章，看了别人写的代码，大多数都是出自一两个人的文章，或者是转载，然后里面就简单的提到了要记录父母的性别，没说具体是什么原因，但是这个时候想必大家都已经绞尽脑汁了，根本不会想到竟然会问父母能不能 “再婚” ，所以我给大家来一个详细的解释，如果还是看不懂的话评论区找我。

        此处的坑是除了已给的性别，每个已知性别的人的父母的性别也是隐含在题中的，每组数据中先是父亲，再是母亲，所以也要把 父亲 母亲 的性别存到sexs字典里，因为会有测试点问你某两个没有性别的人是否能结婚，比如题上的测试案例，如果问 02222 与 03333 是否能结婚，答案是能，因为他们一男一女，并且父母不可考（可视为-1），如果不记录他们的性别的话，在一开始判断性别是否相同的时候就已经进行不下去了，所以会报错。

n = int(input())
a = {}
sexs = {}
for i in range(n):
    id1, sex1, id2, id3 = input().split()
    a[int(id1)] = [int(id1), int(id2), int(id3)]
    # 存本人性别，后面别忘了存父母性别
    sexs[int(id1)] = sex1
    if int(id2) != -1:
        sexs[int(id2)] = 'M'
    if int(id3) != -1:
        sexs[int(id3)] = 'F'

n_love = int(input())
ans = []
p_love = []
for i in range(n_love):
    p_love.append(input())
for i in p_love:
    p1, p2 = map(int, i.split())
    if sexs[p1] == sexs[p2]:
        ans.append("Never Mind")
    else:
        p1_ZuXian = [p1]  # 存p1的祖先
        p2_ZuXian = [p2]  # 存p2的祖先
        used1 = [-1]
        used2 = [-1]
        num = 1
        # 每次循环向上找一层，将自己的祖先放到自己的数组中
        while num < 5:
            num += 1
            # 一定要用长度，如果用 for k in p1_ZuXian: 的话在宾利的
            # 过程中 p1_ZuXian 的长度会变，就控制不住了
            for k in range(len(p1_ZuXian)):  
                if p1_ZuXian[k] in used1 or p1_ZuXian[k] not in a:
                    continue
                used1.append(p1_ZuXian[k])
                p1_ZuXian.append(a[p1_ZuXian[k]][1])
                p1_ZuXian.append(a[p1_ZuXian[k]][2])
            for k in range(len(p2_ZuXian)):
                if p2_ZuXian[k] in used2 or p2_ZuXian[k] not in a:
                    continue
                used2.append(p2_ZuXian[k])
                p2_ZuXian.append(a[p2_ZuXian[k]][1])
                p2_ZuXian.append(a[p2_ZuXian[k]][2])
        # 最后判断两组祖先中有没有共同元素即可
        flag = 1
        for k in p1_ZuXian:
            if k == -1:
                continue
            if k in p2_ZuXian:
                ans.append("No")
                flag = 0
                break
        if flag == 1:
            ans.append("Yes")
for i in ans:
    print(i)