道路与航线题解

farawaytravelerchy

已于 2023-10-05 12:43:55 修改

阅读量77

点赞数 1

分类专栏：蓝书《算法竞赛进阶指南》学习笔记文章标签：算法 python

于 2023-10-05 12:36:14 首次发布

本文链接：https://blog.csdn.net/smile__everydays/article/details/133578674

版权

蓝书《算法竞赛进阶指南》学习笔记专栏收录该内容

5 篇文章 0 订阅

订阅专栏

蓝书最短路例题

题目描述

农夫约翰正在一个新的销售区域对他的牛奶销售方案进行调查。

他想把牛奶送到 T个城镇，编号为 1∼T。

这些城镇之间通过 R 条道路 (编号为 1 到 R) 和 P 条航线 (编号为 1 到 P) 连接。

每条道路 i 或者航线 i 连接城镇 Ai到 Bi，花费为 Ci。

对于道路，0≤Ci≤10,000;然而航线的花费很神奇，花费 Ci 可能是负数(−10,000≤Ci≤10,000)。

道路是双向的，可以从 Ai 到 Bi，也可以从 Bi 到 Ai，花费都是 Ci。

然而航线与之不同，只可以从 Ai 到 Bi。

事实上，由于最近恐怖主义太嚣张，为了社会和谐，出台了一些政策：保证如果有一条航线可以从 Ai 到 Bi，那么保证不可能通过一些道路和航线从 Bi 回到 Ai。

由于约翰的奶牛世界公认十分给力，他需要运送奶牛到每一个城镇。

他想找到从发送中心城镇 S 把奶牛送到每个城镇的最便宜的方案。

输入格式

第一行包含四个整数 T,R,P,S。

接下来 R 行，每行包含三个整数（表示一个道路）Ai,Bi,Ci。

接下来 P 行，每行包含三个整数（表示一条航线）Ai,Bi,Ci。

输出格式

第 1..T1行：第 i行输出从 S 到达城镇 i 的最小花费，如果不存在，则输出 NO PATH。

数据范围

1≤T≤25000
1≤R,P≤50000
1≤Ai,Bi,S≤T

输入样例：

输出样例：

NO PATH
NO PATH
5
0
-95
-100

题目分析(拓扑排序套dijkstra) O(mlogn)

由题意“事实上，由于最近恐怖主义太嚣张，为了社会和谐，出台了一些政策：保证如果有一条航线可以从 Ai 到 Bi，那么保证不可能通过一些道路和航线从 Bi 回到 Ai。” 可知负权边只有可能是有向边并且无环，每一条航线连接两个连通块，对连通块进行拓扑排序，再按拓扑序对连通块里的点跑dijkstra最后可以的到解。具体实现见代码及注释。

python 代码

import sys
import heapq
from queue import Queue

sys.setrecursionlimit(1000000)   #手动加栈
inf = float("inf")
n,r,p,s = map(int, sys.stdin.readline().split())
g = {i:[] for i in range(1,n+1)}
dist = [inf] * (n+1)
vis = [False] * (n+1)
q = Queue()  #用于拓扑排序
color = [0] * (n+1)  #对图染色
node = [[] for _ in range(n+1)]   #第n块有哪些点
deg = [0] * (n+1)

"""建图"""
for _ in range(r):
    u,v,w = map(int, sys.stdin.readline().split())
    g[u].append((v,w,0))     # mark = 0，表示是无向边，再连通块内
    g[v].append((u,w,0))
for _ in range(p):
    u,v,w = map(int, sys.stdin.readline().split())
    g[u].append((v,w,1))     # mark = 1

"""染色"""
def dfs(x):
    color[x] = cnt
    node[cnt].append(x)
    for v,w,z in g[x]:
        if z == 1: continue
        if z == 0:
            if color[v] == 0:
                dfs(v)
cnt = 0   #标记块的号
for i in range(1,n+1):
    if color[i] == 0:
        cnt += 1
        dfs(i)
# print(*color)

"""算度数"""
for x in range(1, n+1):
    for y,w,z in g[x]:
        if z == 1: deg[color[y]] += 1
# print(*deg)

"""拓扑排序套Dijkstra"""
def dijkstra(k):   #对第k块进行dijkstra
    heap = []
    for x in node[k]:    #将所有的第 k 块的入堆
        heapq.heappush(heap, (dist[x], x))
    while heap:
        x = heapq.heappop(heap)
        if vis[x[1]] : continue
        vis[x[1]] = True
        for v,w,z in g[x[1]]:
            if z == 0:   #在内部
                if dist[v] > dist[x[1]] + w:
                    dist[v] = dist[x[1]] + w
                    heapq.heappush(heap, (dist[v], v))
            else:     #在外部，则更新拓扑排序
                deg[color[v]] -= 1
                dist[v] = min(dist[v], dist[x[1]] + w)
                if deg[color[v]] == 0: q.put(color[v])
for i in range(1,cnt+1):
    if deg[i] == 0: q.put(i)    #将所有的0入度的强连通块入队
dist[s] = 0
while not q.empty():
    k = q.get()
    dijkstra(k)
# print(*dist)
for i in range(1,n+1):
    print(dist[i] if dist[i] < n*10000 else "NO PATH")