最短路 迪杰斯特拉 模板 堆优化 python

P3371 【模板】单源最短路径（弱化版）

【模板】单源最短路径（弱化版） - 洛谷

、

题目描述

如题，给出一个有向图，请输出从某一点出发到所有点的最短路径长度。

输入格式

第一行包含三个整数 n,m,s分别表示点的个数、有向边的个数、出发点的编号。

接下来 m 行每行包含三个整数 u,v,w表示一条 u→v 的，长度为 w 的边。

输出格式

输出一行 n 个整数，第 ii 个表示 ss 到第 ii 个点的最短路径，若不能到达则输出 2^31-1。

输入输出样例

输入 

4 6 1
1 2 2
2 3 2
2 4 1
1 3 5
3 4 3
1 4 4

输出 #1

0 2 4 3

O(n^2)模板

n,m,s=map(int,input().split())#有向图
e=[[] for i in range(n+1)]#e[u]表示节点u所有出边终点和边权
dis=[float("inf")]*(n+1) #表示源点到其他点的最短距离，初始化为无穷大
dis[s]=0#源点到源点的距离为0
vis=[0]*(n+1)#表示是否被标记过
vis[s]=1
for i in range(m):
    a,b,c=map(int,input().split())
    e[a].append([b,c])
for i in range(len(e[s])):
    dis[e[s][i][0]]=e[s][i][1]

def dijie(s):#s表示源点
    for i in range(n-1):#表示枚举的次数
        u=0
        for j in range(1,n+1):
            if vis[j]==0 and dis[j]<dis[u]:
                u=j#找到未被标记且离源点最短的下标
        vis[u]=1#找到的点标记
        for k in range(len(e[u])):
            v=e[u][k][0]#点
            w=e[u][k][1]#边权
            if dis[v]>dis[u]+w:
                dis[v]=dis[u]+w
dijie(s)
for i in dis[1:]:
    print(i,end=" ")

O(mlogm)堆优化

from heapq import *#堆优化用到heapq模块
inf=pow(2,31)-1#无穷大
n,m,s=map(int,input().split())
e=[[] for i in range(n+1)]
for i in range(m):
    a,b,c=map(int,input().split())
    e[a].append([b,c])#点a到b的距离为c
dis=[inf]*(n+1)
dis[s]=0#源点到源点距离为0
vis=[0]*(n+1)#是否被标记
heap=[[0,s]]#heap种放源点到其他点的距离和点即[l,u]l是距离，u是点，heappop会自动根据第一个l进行小根堆，l最小的在上面被弹出
while heap:
    l,u=heappop(heap)
    if vis[u]==0:
        vis[u]=1
        for k in e[u]:
            if dis[k[0]]>dis[u]+k[1]:#更新距离
                dis[k[0]]=dis[u]+k[1]
                heappush(heap,[dis[k[0]],k[0]])
for i in range(1,n+1):
    print(dis[i],end=" ")