L2-038 病毒溯源(Python3)

病毒容易发生变异。某种病毒可以通过突变产生若干变异的毒株，而这些变异的病毒又可能被诱发突变产生第二代变异，如此继续不断变化。

现给定一些病毒之间的变异关系，要求你找出其中最长的一条变异链。

在此假设给出的变异都是由突变引起的，不考虑复杂的基因重组变异问题 —— 即每一种病毒都是由唯一的一种病毒突变而来，并且不存在循环变异的情况。

输入格式：

输入在第一行中给出一个正整数 N（≤104），即病毒种类的总数。于是我们将所有病毒从 0 到 N−1 进行编号。

随后 N 行，每行按以下格式描述一种病毒的变异情况：

k 变异株1 …… 变异株k

其中 k 是该病毒产生的变异毒株的种类数，后面跟着每种变异株的编号。第 i 行对应编号为 i 的病毒（0≤i<N）。题目保证病毒源头有且仅有一个。

输出格式：

首先输出从源头开始最长变异链的长度。

在第二行中输出从源头开始最长的一条变异链，编号间以 1 个空格分隔，行首尾不得有多余空格。如果最长链不唯一，则输出最小序列。

注：我们称序列 { a1​,⋯,an​ } 比序列 { b1​,⋯,bn​ } “小”，如果存在 1≤k≤n 满足 ai​=bi​ 对所有 i<k 成立，且 ak​<bk​。

输入样例：

10
3 6 4 8
0
0
0
2 5 9
0
1 7
1 2
0
2 3 1

输出样例：

4
0 4 9 1

提交结果：

代码：

def DFS(s, depth):
    global maxDepth
    global path
    global ans
    if len(child[s]) == 0:
        if depth > maxDepth:
            maxDepth = depth
            ans = path
        return
    for i in range(len(child[s])):
        path.append(child[s][i])
        DFS(child[s][i], depth + 1)
        path = path[:-1]


n = int(input())
father = [-1 for i in range(n)]
child = [[] for i in range(n)]
path = []
ans = []
maxDepth = 0
for i in range(n):
    father[i] = i
for i in range(n):
    data = list(map(int, input().split()))
    for j in range(data[0]):
        child[i].append(data[j + 1])
        father[data[j + 1]] = i
    if len(child[i]) != 0:
        child[i].sort()
root = 0
for i in range(n):
    if father[i] == i:
        root = father[i]
path.append(root)
DFS(root, 1)
print(maxDepth)
for i in range(len(ans)):
    print(ans[i], end="")
    if i != len(ans) - 1:
        print(" ", end="")