作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
又是一道最短路维护额外信息的题目,只需要想清楚什么是时候更新额外信息即可。
cnt[i],路径数量:如果当前的点的最短路被更新了的话,对应这个点的最短路径数量应该是更新点的路径数量。例如,当前点是5,但是最短路被更新了是从3过来的,那么5的最短路径条数应该就是3的最短路径条数。反之,如果最短路径没有被更新而是相等的话,就只需要加上更新点的路径条数即可。例如,本来3到5是最短路径,但是4到5也是最短路径,那么最短路径加上到3的最短路径数量就可以了
r[i]:表示路径上的救援人员总量,首先起点u的救援人员就是对应的w[u]。那么什么时候r会被更新呢?1.当最短路被更新的时候,说明路径都改变了,那么r应该被更新为r[j] = r[t] + w[j],j这个点的救援人员总数应该是从t过来的救援人员总数 + 自己的救援人员。2.最短路出现相等的情况,因为题目要输出的路径是救援人员最多的情况,所以除非另外一条路径上的人数比当前到多,否则不更新路径,也不更新前驱pre
#include <bits/stdc++.h>
using namespace std;
int n, m, S, D;
const int N = 1010;
int g[N][N], w[N];
int st[N];
int dist[N], r[N],path[N];
int cnt[N];
void dijkstra(int u, int s)
{
memset(dist, 0x3f, sizeof dist);
dist[u] = 0;
r[u] = w[u];
for (int i = 0; i < n; i ++ )
{
int t = -1;
for (int j = 0; j < n; j ++ )
if (!st[j] && (t == -1 || dist[t] > dist[j]))
t = j;
st[t] = true;
for (int j = 0; j < n; j ++ )
{
if (dist[j] > dist[t] + g[t][j])
{
dist[j] = dist[t] + g[t][j];
path[j] = t;
r[j] = r[t] + w[j];
cnt[j] = cnt[t];//如果更新了路径,那么这个点的路径数量应该是从起点到更新点的路径数量,因为更新点可以走到j点
}
else if (dist[j] == dist[t] + g[t][j])
{
cnt[j] += cnt[t];
if(r[j] < r[t] + w[j])//题目其实是要求路径最短且人数最多的路径,所以这样才更新前驱
{
r[j] = r[t] + w[j];
path[j] = t;
}
}
}
}
cout << cnt[s] << ' ' << r[s] << endl;
vector<int> q;
int k = s;
q.push_back(k);
while (true)
{
if(k == u) break;
q.push_back(path[k]);
//cout << path[k] << ' ';
k = path[k];
}
for (int i = q.size() - 1; i >= 0; i -- )
if (i != 0) cout << q[i] << ' ';
else cout << q[i] << endl;
}
int main()
{
cin >> n >> m >> S >> D;
for (int i = 0; i < n; i ++ ) cin >> w[i], cnt[i] = 1;//每个点最少1条路径
memset(g, 0x3f, sizeof g);
while (m -- )
{
int a, b, c;
cin >> a >> b >> c;
g[a][b] = g[b][a] = c;
}
dijkstra(S, D);
return 0;
}