作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
经典迪杰斯特拉算法
这题一开始就没仔细读题,要输出的是不同最短路的数量,一开始看成了最短路的长度,没理解road的值遇到更新的情况为什么是road[tp]=road[j],还是要耐心好好读题
AC代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,c1,c2; //c1,c2分别代表起点与终点
const int inf = 999999;
bool visit[520]; //标记该点是否被添加
int dui[520],e[520][520],road[520],duinum[520],dis[520],pre[520]; //dui:当前点救援队数量,e:两点间长度,road:不同最短路数量,duinum:救援队总数,dis:从源点到该点的最短距离,pre:最短路径该点的前驱
void printPath(int v){ //打印最短路径
if(v == c1){
cout << v;
return;
}
printPath(pre[v]);
cout << " " << v;
}
int main(){
cin >> n >> m >> c1 >> c2;
fill(e[0],e[0]+520*520,inf);
fill(dis,dis+520,inf);
for(int i = 0;i < n;i++){
cin >> dui[i];
}
int a,b,c;
for(int i = 0;i < m;i++){
cin >> a >> b >> c;
e[a][b] = c;
e[b][a] = c;
}
dis[c1] = 0; //源点到自己的最短距离就是0
road[c1] = 1; //源点到自己的最短路径就一条
duinum[c1] = dui[c1];
for(int i = 0;i < n;i++){
int tp = -1,min = inf;
for(int j = 0;j < n;j++){ //找到当前的最短路
if(visit[j] == false && dis[j] < min){
min = dis[j];
tp = j;
}
}
if(tp == -1)
break;
visit[tp] = true;
for(int j = 0;j < n;j++){
if(dis[tp] + e[tp][j] < dis[j]){ //可以更新长度
dis[j] = dis[tp]+e[tp][j];
duinum[j] = duinum[tp] + dui[j];
road[j] = road[tp]; //只加了一条边,不同最短路的数量不变
pre[j] = tp;
}else if(dis[tp] + e[tp][j] == dis[j]){
road[j] += road[tp]; //遇到相等的情况,不同最短路的数量增加
if(duinum[j] < duinum[tp] + dui[j]){
duinum[j] = duinum[tp] + dui[j];
pre[j] = tp;
}
}
}
}
cout << road[c2] << " " << duinum[c2] << endl;
printPath(c2);
}