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=" ")