L2-001 紧急救援 (25分)
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
解题
题目要求得出单源最短路径,但还需算出最短路径条数,最短路径中最优顶点权重,借助 num[ ] 数组记录每个顶点的最短路径条数,ans[ ] 数组记录最优顶点权重,num[ ] 数组在最短路取等号时加数,具体实现看代码
代码
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
const int maxn = 5e2 + 2;
int G[maxn][maxn]; // 边权重
int V[maxn]; // 顶点权重
int dist[maxn], path[maxn];
int num[maxn]; // 记录最短路条数
int ans[maxn]; // 记录最优顶点权重
bool vis[maxn];
struct node{
int v;
int w;
};
struct cmp{
bool operator () (const node &a, const node &b){
return a.w > b.w;
}
};
void Dijkstra(int S, int D, int N){
memset(dist, 0x3f, sizeof(dist));
memset(path, -1, sizeof(path));
dist[S] = 0;
ans[S] = V[S];
num[S] = 1;
vis[S] = true;
priority_queue<node, vector<node>, cmp> q;
for(int i = 0; i < N; i++){
if(!vis[i] && G[S][i]){
q.push({i, G[S][i]});
dist[i] = G[S][i];
path[i] = S;
num[i] = num[S];
ans[i] = ans[S] + V[i];
}
}
while(!q.empty()){
int v = q.top().v;
int w = q.top().w;
q.pop();
if(vis[v]) continue;
vis[v] = true;
for(int i = 0; i < N; i++){
if(!vis[i] && G[v][i]){
if(dist[v] + G[v][i] < dist[i]){
dist[i] = dist[v] + G[v][i];
q.push({i, dist[i]});
num[i] = num[v]; // 单路径时, 接上一个
ans[i] = ans[v] + V[i];
path[i] = v;
} else if(dist[v] + G[v][i] == dist[i]){
num[i] += num[v]; // 重复路径时, 增加
// ans[i] = max(ans[i], ans[v] + V[i]);
if(ans[i] < ans[v] + V[i]){
ans[i] = ans[v] + V[i];
path[i] = v;
}
}
}
}
if(v == D) return;
}
}
void Print(int D, int S){ // 输出最优解
if(path[D] == -1 && D == S){
cout << D;
return;
}
Print(path[D], S);
cout << " " << D;
}
int main() {
int N, M, S, D;
cin >> N >> M >> S >> D;
for(int i = 0; i < N; i++) cin >> V[i];
int v1, v2, w;
for(int i = 0; i < M; i++){
cin >> v1 >> v2 >> w;
G[v1][v2] = G[v2][v1] = w;
}
Dijkstra(S, D, N);
cout << num[D] << " " << ans[D] << endl;
// for(int i = 0; i < N; i++){
// cout << dist[i] << " " << num[i] << " " << ans[i] << endl;
// }
Print(D, S);
return 0;
}