作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
#include <iostream>
#include <cstring>
using namespace std;
int n, m, c1, c2;
int f[510]; // 记录途径那些点
int dist[510], cn[510], sum[510], rshu[510];
int mp[510][510];
bool st[510];
void print(int c2, int c1)
{
if (c2 == c1)
{
cout << c2; // 终点就是起点,最后会递归到这一步再开始回溯
return;
}
print(f[c2], c1); // f[]最后是最后的点,所以得倒着找点
cout << " " << c2;
}
void dijkstra()
{
memset(dist, 0x3f, sizeof dist);
dist[c1] = 0, cn[c1] = 1, sum[c1] = rshu[c1]; // 确定起始顶点编号,条数(起点到其本身方式为1),人数
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] = 1;
for (int j = 0; j < n; j++) // 确定t后更新各个点的最短路
// dist[j]=min(dist[j],dist[t]+mp[t][j]);
{
if (dist[j] > dist[t] + mp[t][j])
{
f[j] = t; // j的上一个点是t
dist[j] = dist[t] + mp[t][j];
// 从起点走到路径j的数量,j的最短路径条数得算上j的前一个点的最短路径条数(方式等于到达前驱的方式)
cn[j] = cn[t];
sum[j] = sum[t] + rshu[j];
}
else if (dist[j] == dist[t] + mp[t][j]) // 如果两者最短路一样
{
if (sum[j] < sum[t] + rshu[j])
{
f[j] = t; // j的上一个点是t
sum[j] = sum[t] + rshu[j];
}
cn[j] = cn[t] + cn[j]; // 合并路径数量,两种方式距离相等,则方式数相加
// sum[j]=max(sum[j],sum[t]+rshu[j]); //比较人数,选那个人多的点u
}
}
}
}
int main()
{
cin >> n >> m >> c1 >> c2;
for (int i = 0; i < n; i++)
cin >> rshu[i];
memset(mp, 0x3f, sizeof mp);
int x, y, d;
for (int i = 0; i < m; i++)
{
cin >> x >> y >> d;
mp[x][y] = mp[y][x] = min(mp[x][y], d); // 无向图对称 不写这个12分
}
dijkstra();
cout << cn[c2] << " " << sum[c2] << endl;
print(c2, c1);
return 0;
}