L2-001 紧急救援 (25 分)(Dijkstra记录)
程序员文章站
2024-02-24 17:38:43
...
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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 上想,然后分析题意,。
第一步:他是一个Dijkstra算法,
第二步:题目要求,求出共有几条最短路,所以我们应该记录源点到某个点有几条最短路,
第三步:我们在找出最短路的情况下,找到最多的救援队,
第四步:路径输出,
怎么用Dijkstra 写呢?
我们用一个点更新时 判断是不是最短路 如果是 直接更新, 如果距离相等,就需要判断走哪条路的救援队最多
别忘了记录路径呦。
#include<iostream>
#include<cstring>
using namespace std;
const int INF = 0x3f3f3f3f;
void pathPrintf(int path[], int end){
int s[520];
int top = -1;
int temp = path[end];
s[++top] = end;
while(temp != -1){
s[++top]= temp;
temp = path[temp];
}
while( top != -1){
cout << s[top--] << " ";
if(top == 0)
cout << s[top--];
}
}
int main(){
int N, M, S, D;
cin >> N >> M >> S >> D;
int mp[505][505];
int help[505][3];
int num [505];
bool vis[505];
int path[505];
int dis[505];
memset(dis, INF, sizeof(dis));
memset(vis, 0, sizeof(vis));
memset(mp, INF, sizeof(mp));
memset(help, 0, sizeof(help));
memset(num, 0, sizeof(num));
memset(path, -1, sizeof(path));
for(int i = 0; i < N; i++){
cin >> help[i][0];
}
for(int i = 0; i < M; i++){
int x, y, z;
cin >> x >> y >> z;
mp[x][y] = z;
mp[y][x] = z;
}
for(int i = 0; i < N; i++){
dis[i] = mp[S][i];
if(dis[i] != INF){
help[i][1] = help[S][0] + help[i][0];
path[i] = S;
num[i] = 1;
}
}
vis[S] = 1;
path[S] = -1;
dis[S] = 0;
num[S] = 1;
help[S][1] = help[S][0];
while(1){
int k = -1;
int minn = INF;
for(int i = 0; i < N; i ++){
if( !vis[i] && minn > dis[i] ){
minn = dis[i];
k = i;
}
}
if(k == -1)
break;
vis[k] = 1;
for(int i = 0; i < N; i++){
if( !vis[i] && dis[i] > dis[k] + mp[k][i] ){
dis[i] = dis[k] + mp[k][i]; //更新距离
path[i] = k; //更新上个结点
help[i][1] = help[k][1] + help[i][0]; //更新救援队数量
num[i] = num[k]; //更新有几条最短路
}
else if(!vis[i] && dis[i] == dis[k] + mp[k][i]){
num[i] = num[k] + num[i]; //更新有几条路
if(help[i][1] < help[k][1] + help[i][0]){ //判断救援队是否变多
help[i][1] = help[k][1] + help[i][0]; // 如果变多 则加上
path[i] = k; //改上个结点的值
}
}
}
}
cout << num[D] << " " << help[D][1] << endl;
pathPrintf(path, D);
}