L2-001 紧急救援（25 分）(Dijkstra求最短路)

最新推荐文章于 2024-06-24 12:17:25 发布

原创最新推荐文章于 2024-06-24 12:17:25 发布 · 424 阅读

0 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一个基于Dijkstra算法的紧急救援路线寻找方案，通过优化最短路径和计算最多可召集的救援队数量，实现了在复杂城市网络中快速响应并调动资源的目标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目】

L2-001 紧急救援（25 分）

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式:

输入第一行给出4个正整数N、M、S、D，其中N（2≤N≤500）是城市的个数，顺便假设城市的编号为0 ~ (N−1)；M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。

第二行给出N个正整数，其中第i个数是第i个城市的救援队的数目，数字间以空格分隔。随后的M行中，每行给出一条快速道路的信息，分别是：城市1、城市2、快速道路的长度，中间用空格分开，数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式:

第一行输出最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔，输出结尾不能有多余空格。

输入样例:

4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

输出样例:

2 60
0 1 3

【题解】

思路：Dijkstra算法求最短路的过程中更新最短路径和最短路径的条数和最多集结救援队的数目。

【代码】

#include <bits/stdc++.h>
using namespace std;
const int inf = 0x3f3f3f3f;
int mp[505][505];
int vis[505]={0};  //是否被选中
int cou[505]={0};  //有多少条最短路径
int sum[505]={0};  //最多能组织的救援队数目
int pre[505];
int a[505];
int n,m,s,d;
void dij()
{
    int dis[505];
    for(int i=0;i<n;i++){
        dis[i]=mp[s][i];
        if(dis[i]<inf){
            sum[i]=a[s]+a[i];
            cou[i]=1;
            pre[i]=s;
        }
    }
    vis[s]=1;
    sum[s]=a[s];
    for(int i=1;i<n;i++){
        int next=-1;
        int mi=inf;
        for(int j=0;j<n;j++){
            if(!vis[j]&&dis[j]<mi){
                mi=dis[j];
                next=j;
            }
        }
        if(next==-1) continue;
        vis[next]=1;
        for(int j=0;j<n;j++)
            if(!vis[j])
            {
                if(dis[next]+mp[next][j]==dis[j]){
                    cou[j]+=cou[next];
                    if(sum[next]+a[j]>sum[j]){
                        sum[j]=sum[next]+a[j];
                        pre[j]=next;
                    }

                }
                else if(dis[next]+mp[next][j]<dis[j]){
                    dis[j]=dis[next]+mp[next][j];
                    cou[j]=cou[next];
                    sum[j]=sum[next]+a[j];
                    pre[j]=next;
                }
            }
    }
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&s,&d);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            if(i==j) mp[i][j]=0;
            else mp[i][j]=inf;
    while(m--){
        int a,b,v; scanf("%d%d%d",&a,&b,&v);
        mp[a][b]=mp[b][a]=min(mp[a][b],v);
    }
    dij();
    printf("%d %d\n",cou[d],sum[d]);
    stack <int> stk;
    while(d!=s){
        stk.push(d);
        d=pre[d];
    }
    printf("%d",s);
    while(!stk.empty()){
        printf(" %d",stk.top());
        stk.pop();
    }
    puts("");
    return 0;
}