【天梯】python L2-001 城市间紧急救援 (25 point(s))

原创已于 2022-04-13 09:28:37 修改 · 2k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法

于 2022-04-13 09:28:00 首次发布

PAT 同时被 2 个专栏收录

19 篇文章

订阅专栏

天梯

18 篇文章

订阅专栏

该博客探讨了图论中的最短路径算法，并扩展到同时考虑最短路径条数和最大资源的问题。通过Dijkstra算法求解单源最短路径，并在路径中累积资源。最终输出从起点到终点的最短路径条数和最大资源总量。此外，还提供了一个回溯函数用于构造最短路径。

在这里插入图片描述
救命这个也超时

1.最短路径
2.最短路径条数
3.最大资源

n,m,s,d=map(int,input().split())
x=list(map(int,input().split()))#资源
city=[[float('inf') for i in range(n)]for j in range(n)]
dis=[float('inf') for i in range(n)]
vis=[-1 for i in range(n)]#记录顶点
num=[0 for i in range(n)]#最短路径条数
w=[0 for i in range(n)]#最大资源
pre=[i for i in range(n)]#记录前驱

for i in range(m):
    a,b,c=map(int,input().split())
    city[a][b]=c
    city[b][a]=c
def dijkstra(s):
    dis[s]=0
    num[s]=1
    w[s]=x[s]
    for i in range(n-1):
        mi=float('inf')
        u=0
        for j in range(n):
            if vis[j]==-1 and  dis[j]<mi:
                mi=dis[j]
                u=j
        if u==-1:
            return
        vis[u]=1#访问距离最近的点
        for v in range(n):#更新前驱
            if vis[v]==-1:                
                if dis[v]>dis[u]+city[u][v]:
                    dis[v]=dis[u]+city[u][v]
                    num[v]=num[u]
                    w[v]=w[u]+x[v]
                    pre[v]=u
                elif dis[v]==dis[u]+city[u][v]:
                    num[v]+=num[u]
                    if(w[v]<w[u]+x[v]):
                        w[v]=w[u]+x[v]
                        pre[v]=u
dijkstra(s)
print("%d %d"%(num[d],w[d]))
k=[]
def dfs(s,d):
    if s==d:
        k.append(s)
        return
    dfs(s,pre[d])
    k.append(d)
dfs(s,d)
print(*k)