美团编程——病毒传播

题目描述

给出一个图G(V,E)，图上有n个点，m条边，所有的边都是无向边。

最开始，也就是第0天的时候，这n个点中有一个点v感染了病毒，之后的每一天，凡是感染病毒的点都会向它的邻居点传播病毒。经过了t天之后，得到了感染病毒的点集S。要求找出第0天感染病毒的点v。如果v有很多不同的答案，把它们都找出来。

输入描述:

输出描述: 

示例

输入	输出
4 3 3 2 1 2 1 4 3 2 4 2 1	4

 说明

第0天，第1天，第2天感染病毒的点如图

思路：

根据广度优先搜索算法从起始结点开始访问，进行深度为病毒传播天数的访问，

代码：

n, m = [int(x) for x in input().split()]
vers = [[] for i in range(n)]
for i in range(m):
    u, v = [int(x) - 1 for x in input().split()]
    vers[u].append(v)
    vers[v].append(u)

k, t = [int(x) for x in input().split()]
data = [int(x) - 1 for x in input().split()]
#print(vers)
data.sort()
v_true = []
def bfs(index,t):
    visted = [0 for i in range(n)]
    visted[index]=1
    queue=[]
    queue.append([index,0])
    while queue:
        v=queue.pop(0)
        if v[1]<t:
            for i in vers[v[0]]:
                if visted[i]==0:
                    queue.append([i,v[1]+1])
                    visted[i]=1
    return visted
results = []
for v in range(n):
    result = []
    visited=bfs(v,t)
    #print(v,visited)
    for i,w in enumerate(visited):
        if w==1:
            result.append(i)
    if data == result:
        results.append(str(v + 1))
if len(results) > 0:
    print(" ".join(results))
else:
    print("-1")