题目描述
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
分析
看到各点路径大于0,求最端路径。第一个想到的就是dijkstra算法,由于N很小,不需要堆优化,直接用邻接矩阵。至于题目中的其他要求输出,在dijkstra中加入一些数组转化就好。具体看代码注解。
#include <bits/stdc++.h>
using namespace std;
const int N = 510;
int n, m, s, d;
int dist[N], g[N][N], fireCnt[N], pathCnt[N], sumFire[N], pre[N];
bool st[N];
/*
dist[N] s到各点的距离
g[N][N] a, b两点的距离
fireCnt[N] 各城市消防队数量
pathCnt[N] s到各点的最近距离情况的路径数
sumFire[N] s到各点的最近距离情况下,所有路径中经过城市消防数量的总和最大值
pre[N] 最优情况下,点a的上一个节点
st[N] dijkstra中各个城市的是否已经达到最小路径的标记。
*/
void dijkstra(int s, int d) {
memset(dist, 0x3f, sizeof dist);
dist[s] = 0; //初始化
pathCnt[s] = 1;
sumFire[s] = fireCnt[s];
//选出当先s点可以到达距离最近城市(还未被选择的城市)
for (int i = 0; i < n; i++) {
int k = -1;
for (int j = 0; j < n; j++) {
if (!st[j] && (k == -1 || dist[k] > dist[j]))
k = j;
}
st[k] = true; //一旦选出,那么dist[k]这个距离就是到k的最小距离。
// 通过disk[k]更新各点的dist
//(因为dist[k]已经是目前情况下,最小路径,只需更新还未被选的城市)
for (int j = 0; j < n; j++) {
if (dist[j] == dist[k] + g[k][j]) {
pathCnt[j] += pathCnt[k]; //两种不同路径的距离一致,迭代更新
if (sumFire[j] < sumFire[k] + fireCnt[j]) {
sumFire[j] = sumFire[k] + fireCnt[j]; //选择消防数量大的路径更新
pre[j] = k; //更新前一个节点
}
}
if (dist[j] > dist[k] + g[k][j]) {
dist[j] = dist[k] + g[k][j];
pathCnt[j] = pathCnt[k]; //路径距离小,更新为上一节点的路径数。
sumFire[j] = sumFire[k] + fireCnt[j]; //不管之前消防数量多大,直接更新
pre[j] = k; //距离小是最前提条件
}
}
}
}
int main() {
cin >> n >> m >> s >> d;
for (int i = 0; i < n; i++) scanf("%d", &fireCnt[i]);
memset(g, 0x3f, sizeof g); //赋值邻接矩阵
while (m --) {
int a, b, w;
scanf("%d%d%d", &a, &b, &w);
g[a][b] = min(g[a][b], w); //防止出现重路径
g[b][a] = min(g[b][a], w);
}
dijkstra(s, d);
cout << pathCnt[d] << " " << sumFire[d] << endl;
int t = d;
stack<int> sk; //因为pre得到的路径是从d到s
while (t != s) { //用各堆将路径倒过来,输出
sk.push(t);
t = pre[t];
}
sk.push(s);
while (sk.size()>1) {
cout << sk.top() << " ";
sk.pop();
}
cout << d;
return 0;
}