L2-001. 紧急救援---(Dijkstra,记录路径)

时间:2023-05-23 20:29:08

https://www.patest.cn/contests/gplt/L2-001

L2-001. 紧急救援

时间限制
200 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越

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

输入格式:

输入第一行给出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 <cstdio>
#include <algorithm>
using namespace std;
int n, m, c1, c2;
int dis[], weight[], e[][];///dis从出发点到i点最短路径,weight对应救援队,e图
int num[], w[], pre[];///num从出发点到i的路径条数,w救援队数目,i的前一个结点
bool visit[];
const int inf = 0x3f3f3f3f;
void printPath(int v) {
if(v == ) {
printf("%d", v);
return ;
}
printPath(pre[v]);
printf(" %d", v);
}
int main() {
scanf("%d%d%d%d", &n, &m, &c1, &c2);
for(int i = ; i < n; i++)
scanf("%d", &weight[i]);
fill(e[], e[] + * , inf);
fill(dis, dis + , inf);
int a, b, c;
for(int i = ; i < m; i++) {
scanf("%d%d%d", &a, &b, &c);
e[a][b] = c;
e[b][a] = c;
}
dis[c1] = ;
w[c1] = weight[c1];
num[c1] = ;
for(int i = ; i < n; i++) {
int u = -, minn = inf;
for(int j = ; j < n; j++) {
if(visit[j] == false && dis[j] < minn) {
u = j;
minn = dis[j];
}
}
if(u == -) break;
visit[u] = true;
for(int v = ; v < n; v++) {
if(visit[v] == false && e[u][v] != inf) {
if(dis[u] + e[u][v] < dis[v]) {///无条件选择距离短的
dis[v] = dis[u] + e[u][v];///长度
num[v] = num[u];
w[v] = w[u] + weight[v];///资源
pre[v] = u;
} else if(dis[u] + e[u][v] == dis[v]) {
num[v] = num[v] + num[u];///
if(w[u] + weight[v] > w[v]) {
w[v] = w[u] + weight[v];///救援队尽可能多
pre[v] = u;
}
}
}
}
}
printf("%d %d\n", num[c2], w[c2]);
printPath(c2);///路径上几个点
return ;
}