PTA L2-001 紧急救援

相逢亦逢春

于 2024-04-18 17:06:17 发布

阅读量1.8k

点赞数 37

文章标签：图论算法

本文链接：https://blog.csdn.net/2301_80034662/article/details/137928247

版权

文章讲述了如何利用Dijkstra算法解决城市应急救援问题，涉及计算最短路径以及在路径相同情况下选择召集最多救援队的方法。算法通过优先队列和路径记录数组来找到从出发地到目的地的最短路径，并在路径长度相等时依据救援队数量决定最优路线。

摘要由CSDN通过智能技术生成

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

输入格式:

输入第一行给出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

代码实现:

#include<bits/stdc++.h>
using namespace std;
int N,M,S,D;
typedef pair<int,int> PII;
int m[501][501]; //用来初始化城市之间的给出的最初距离
int path[501],city[501],dis[501],per[501],vis[501],road[501];
//path是用来统计经过的路径 city统计每个城市的救援队数量
//dis 用来更新到达每个城市的最短距离 per用来统计到达每个城市最短路径的同时最多召集的人数
//vis 用来判断此城市是否遍历过 road统计达到某个城市的最短路径数
void meth(){
memset(dis,0x3f3f3f,sizeof(dis)); //初始化
priority_queue<PII> q;
path[S]=-1;
road[S]=1;
per[S]=city[S]; //出发点自带的救援队数量
dis[S]=0; //出发点距离出发点距离为0
q.push({ 0,S});
while(q.size()){
int now=q.top().second;
q.pop();
if(vis[now])
continue;

vis[now]=1;

for(int i=0;i<N;i++){

if(dis[i]>dis[now]+m[now][i]){ //距离需要更新的情况
dis[i]=dis[now]+m[now][i];
path[i]=now;
road[i]=road[now];
per[i]=per[now]+city[i];
if(i!=D){
q.push({-dis[i],i});
}
}else if(dis[i]==dis[now]+m[now][i]){ //两种情况距离相等
if(per[i]<per[now]+city[i]){ //判断救援队数量是否需要更新
per[i]=per[now]+city[i];
path[i]=now;
}
road[i]+=road[now]; //最短路径数量的更新
}
}
}
}
int main(){
cin>>N>>M>>S>>D;
int a,b;
for(int i=0;i<N;i++){
cin>>city[i];
}
memset(m,600,sizeof(m)); //因为给出的数据范围是2-500，所以初始化数据用600就好
for(int i=0;i<M;i++){
cin>>a>>b;
cin>>m[a][b];
m[b][a]=m[a][b];
}
meth();
vector<int> pp; //
int qq=D;
while(qq!=-1){ //将起点到终点的经过路径放进去
pp.push_back(qq);
qq=path[qq];
}
cout << road[D] << " " << per[D] << endl;
for (int i = (int)pp.size() - 1; i > 0; i--)
cout << pp[i] << " ";
cout << pp[0];
return 0;
}