PAT L2-1. 紧急救援 (Dijkstra扩展)

题目链接：https://www.patest.cn/contests/gplt/L2-1

L2-1. 紧急救援

时间限制

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的理解。 具体见代码注释。 有空用优先队列写一发，今天有点头痛....回寝室了

参考博客：/article/7759280.html

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int inf = 0x3f3f3f3f;
int N, M, S, D;
int map[550][550];
int val[550], dis[550], vis[550], cnt[550], sum[550], pre[550];
//每点权值   距S距离   标记      路径数量  最大救援队  前驱
void path(int d) {
if(pre[d] != -1) {
path(pre[d]);
printf("%d ", pre[d]);
}
}
int main() {
scanf("%d %d %d %d", &N, &M, &S, &D);
int i, j;
for(i = 0; i < N; i++) {
scanf("%d", val + i);
}
for(i = 0; i < N; i++) {
for(j = 0; j < N; j++) {
if(i == j) map[i][j] = 0;
else map[i][j] = inf;
}
}
int x, y, z;
for(i = 0; i < M; i++) {
scanf("%d %d %d", &x, &y, &z);
map[x][y] = z;
map[y][x] = z;
}
for(i = 0; i < N; i++) {  //初始化
dis[i] = inf;
vis[i] = 0;
cnt[i] = 0;
sum[i] = 0;
pre[i] = -1;
}
dis[S] = 0;  //初始化起始点
vis[S] = 1;
cnt[S] = 1; //起点到其本身方式为1
sum[S] = val[S];
for(i = 0; i < N; i++) {
int minn = inf, mini = S;
for(j = 0; j < N; j++) {
if(vis[j] == 0 && minn > dis[j]) {
minn = dis[j];
mini = j;
}
}
vis[mini] = 1;
for(j = 0; j < N; j++) {
if(vis[j] == 0) {
if(dis[j] > dis[mini] + map[mini][j]) {
dis[j] = dis[mini] + map[mini][j];
sum[j] = sum[mini] + val[j];
cnt[j] = cnt[mini];  //方式等于到达前驱的方式
pre[j] = mini;
}
else if(dis[j] == dis[mini] + map[mini][j]) {
cnt[j] = cnt[mini] + cnt[j]; //两种方式距离相等，则方式数相加
if(sum[j] < sum[mini] + val[j]) {  //能获得更大的救援，则更新
sum[j] = sum[mini] + val[j];
pre[j] = mini;
}
}
}
}
}
printf("%d %d\n", cnt[D], sum[D]);
path(D);
printf("%d\n", D);
return 0;
}