广度优先搜索：使用 Python 的最短路径

Question

我在 Hackerrank 中遇到了问题Breadth First Search: Shortest Reach，这是我在 python 中的解决方案。但是我的代码只通过了 7 个测试用例中的一个。我在这个实现中哪里错了？

#Hackerrank
#Topic : Graphs
#Url : https://www.hackerrank.com/challenges/bfsshortreach/problem

def bfs(i, n, E):

    level = [-1] * (n+1)
    queue = []
    level[i] = 0
    queue.append(i)

    while len(queue) > 0:
        head = queue[0]
        queue = queue[1:]
        for j,k in E:
            if j == head:
                if level[k] == -1:
                    level[k] = 1 + level[j]
                    queue.append(k)

    return level


q = int(input())
for case in range(q):
    e = input().split(" ")
    n = int(e[0])
    m = int(e[1])
    E = []

    for edge in range(m):
        e = input().split(" ")
        u = int(e[0])
        v = int(e[1])
        E.append([u,v])

    s = int(input())
    distance = bfs(s, n, E)
    dist = []
    for i in distance[1:]:
        if i > 0:
            dist.append(str(i*6))
        if i < 0:
            dist.append(str(-1))

    print(" ".join(dist))
    print("")

Answer 1

问题表明图是无向的，但您只需添加 1 条边即可使其有向：

E.append([u,v])

您还应该添加其他方式的边缘

E.append([v,u])