L2-031 深入虎穴(Python3)

著名的王牌间谍 007 需要执行一次任务，获取敌方的机密情报。已知情报藏在一个地下迷宫里，迷宫只有一个入口，里面有很多条通路，每条路通向一扇门。每一扇门背后或者是一个房间，或者又有很多条路，同样是每条路通向一扇门…… 他的手里有一张表格，是其他间谍帮他收集到的情报，他们记下了每扇门的编号，以及这扇门背后的每一条通路所到达的门的编号。007 发现不存在两条路通向同一扇门。

内线告诉他，情报就藏在迷宫的最深处。但是这个迷宫太大了，他需要你的帮助 —— 请编程帮他找出距离入口最远的那扇门。

输入格式：

输入首先在一行中给出正整数 N（<105），是门的数量。最后 N 行，第 i 行（1≤i≤N）按以下格式描述编号为 i 的那扇门背后能通向的门：

K D[1] D[2] ... D[K]

其中 K 是通道的数量，其后是每扇门的编号。

输出格式：

在一行中输出距离入口最远的那扇门的编号。题目保证这样的结果是唯一的。

输入样例：

13
3 2 3 4
2 5 6
1 7
1 8
1 9
0
2 11 10
1 13
0
0
1 12
0
0

输出样例：

12

提交结果：

代码（测试点3，测试点6非零返回）：

import sys


def dfs(x, d):
    global ans
    global D
    if d >= D:
        D = d
        ans = x
    for i in range(len(v[x])):
        dfs(v[x][i], d + 1)


n = int(sys.stdin.readline())
ans = 0
D = 0
ru = 0
flag = [0 for i in range(n + 1)]
v = [[] for i in range(n + 1)]
for i in range(1, n + 1):
    data = list(map(int, sys.stdin.readline().split()))
    for x in data[1:]:
        flag[x] = 1
        v[i].append(x)
for i in range(1, n + 1):
    if flag[i] == 0:
        ru = i
        break
dfs(ru, 0)
print(ans)