题解 | #【模板】单源最短路1# Python3

【模板】单源最短路1

https://www.nowcoder.com/practice/f24504c9ff544f7c808f76693cc34af8

import sys
from collections import defaultdict
import heapq

n, m = list(map(int,input().strip().split(' ')))

adjacent_mat = defaultdict(list)

min_dis = [5001] * (n+1) 
not_processed = set()

for i in range(m):
    u, v = list(map(int,input().strip().split(' ')))
    adjacent_mat[u].append(v)
    adjacent_mat[v].append(u)

processed = set()
pq = [] # 记录当前所有从起始点到点i还未确定最短路径的距离以及点i

heapq.heappush(pq,(0,1)) # 起始点为1,距离为0

while len(pq):
    cur_dis, cur_pos = heapq.heappop(pq)
    processed.add(cur_pos)

    if cur_pos == n:
        print(cur_dis)
        break
    
    for v in adjacent_mat[cur_pos]:
        if v in processed: continue
        heapq.heappush(pq,(cur_dis+1,v))
else:
    print(-1)

    

    

        





    





全部评论

相关推荐

用微笑面对困难:只要你保证项目和获奖都是真的就行尤其是“对战,总负责人”啊这些套职,基本上队员,打杂的都这么写
点赞 评论 收藏
分享
09-10 14:27
中南大学 营销
谁知道呢_:上过学的就是不一样
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务