spfa求最短路

n,m = map(int,input().split())
prices = [0] + list(map(int,input().split()))
from collections import *
gl = defaultdict(list)
gr = defaultdict(list)
for _ in range(m):
    u,v,t = map(int,input().split())
    gl[u].append(v)
    gr[v].append(u)
    if t == 2:
        gl[v].append(u)
        gr[u].append(v)
        
from math import inf

def spfa(s,g,flag):
    dis = [0]*(n + 1)
    if flag:
        dis = [inf]*(n + 1)
    st = [0]*(n + 1)
    st[s] = True
    q = [s]
    dis[s] = prices[s]
    
    while q:
        u = q.pop()
        st[u] = 0 # 不在队列中
        for v in g[u]:
            if (flag and dis[v] > min(dis[u],prices[v]) ) or (not flag and dis[v] < max(dis[u],prices[v])):
                dis[v] = min(dis[u],prices[v]) if flag else max(dis[u],prices[v])
                # 这里一定注意,有更新,才入队
                if not st[v]:
                    st[v] = 1
                    q.append(v)
    return dis


ans = 0
d_min = spfa(1,gl,True)
d_max = spfa(n,gr,False)
for i in range(1,n + 1):
    ans = max(ans,d_max[i] - d_min[i])
print(ans)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值