作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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;
}
这道题主要是运用了最短路径算法中的Dijkstra算法,与传统的Dijkstra算法用所不同的是你需要一个数组对你经过 的最短距离的路径进行记录,同时出现最短距离相等的情况下,你需要进行对救援队数量的比较,留下数量更多的情况存储在数组当中