作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
思路:
学期初,我准备做完天梯赛的题,但遇到这个图论的题就很头疼,板子题套一下就能过,但这道题有很多拓展,于是我不会写,找别人的代码看,结果看不懂,因为毕竟每个人代码习惯不一样,但今天我自己把做出来了,这里分享一下题的解法和我学习完能做出它的过程。
这是一道最短路问题,有需要额外维护的信息:最大队伍数,最短路径数,最短路径。
刚学完最短路算法,一上来让我维护这么多信息,脑容量真的不够用呀。
为了做出这道题,我们首先要会最短路算法,由于边权不是1,用dijkstra更好,
然后我们要会最短路计数,这个是bfs求最短路,不过思路可以理解
输出最短路径其实是一种思想,就是我每个点,在确定了最短路之后,我是知道这个点是从哪个点来的,这样每个点都保存自己的前驱,那么倒着输出就是这个最短路径了。
最后解决要维护最大救援队的问题:
什么时候需要判断救援队数量会不会更大呢?
当我们这个点有两条最短路(这个点到起点的距离等于dijkstra中用来更新别人的点 t 到起点的距离加上 t 到这个点的距离)的时候(一定要先把基本的最短路算法搞清楚再来研究这个),我们需要判断,这两个最短路,哪个队伍数更多,保存多的那一个即可,注意,这时候这个点的前驱要保存对于数更多的那个,有种最短路套一个最短路的感觉,大家可以体会一下。
代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 505;
int n,m,s,d;
int v[N];
//每个点的队伍数
int g[N][N];
//邻接矩阵存图
int dist[N], path[N],cnt[N],team[N];
//dist 到起点的距离
//path 每个点的前驱
//cnt 每个点的最短路条数
//team 在最短路中每个点的队伍数
bool st[N];
//dijkstra的判断数组
void dijkstra() {
//初始化距离和最短路条数
memset(dist, 0x3f, sizeof dist);
dist[s] = 0;
cnt[s] = 1;
for (int i = 0; i < n - 1; i++) {
int t = -1;
for (int j = 0; j < n; j++)
if (!st[j] && (t == -1 || dist[t] > dist[j]))
t = j;
for (int j = 0; j < n; j++) {
if (dist[j] > dist[t] + g[t][j]) {
//更新最短路
dist[j] = dist[t] + g[t][j];
cnt[j] = cnt[t];//最短路数等于上一个点的最短路数
team[j] = team[t] + v[j];//队伍数等于上一个点的队伍数加这个点的
path[j] = t;//更新了最短路就要改这个点的前驱
}
else if (dist[j] == dist[t] + g[t][j]) {
//如果出现两条最短路
cnt[j] += cnt[t];//最短路数加上一个点的最短路数
if (team[j] < team[t] + v[j]) {
//如果队伍数没有从t点过来的队伍数多
team[j] = team[t] + v[j];
//更新最短路
path[j] = t;
}
}
}
st[t] = true;
}
}
int main()
{
cin >> n >> m >> s >> d;
for (int i = 0; i < n; i++) {
cin >> v[i];
team[i] = v[i];
}
memset(g, 0x3f, sizeof g);
for (int i = 0; i < n; i++)path[i] = i;
while (m--) {
int a, b, c;
cin >> a >> b >> c;
g[a][b] = g[b][a] = min(c, g[a][b]);
}
dijkstra();
cout << cnt[d] << " " << team[d] << endl;
vector<int> res;
res.push_back(d);
for (int i = path[d]; i != s; i = path[i])res.push_back(i);
res.push_back(s);
for (int i = res.size() - 1; i >= 0; i--) {
if (i)cout << res[i] << " ";
else cout << res[i];
}
return 0;
}