作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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
算法思路描述
本题的知识点是dijkstra算法,大概题意为从给出的起始城市到终点城市,首先要找到最短路径,在此基础上然后使救援队的数量最大化。
1.我们需要在更新路径的时候,同时更新救援队的数量和最短路有几条;
2.题目中要求输出通过的城市编号,我们可以开一个数组同时记录这个点的前驱是哪个点;
3.当发现两个路径距离相同时,我们应该更新救援队的数量,只有当当前救援队的数量小于新的路径上的救援队的数量时,再更新最大救援队数量,同时记录该点的前驱。
#include<bits/stdc++.h> using namespace std; const int N=510; int g[N][N],dist[N],city[N]; int cnt[N],maxnum[N],pre[N]; //cnt[N]表示几条最短路,city[N]表示每个城市的救援队数量 //maxnum[N]表示该城市救援队的最大数量; bool st[N]; int n,m,s,d; void dijkstra() { memset(dist,0x3f,sizeof dist); dist[s]=0; cnt[s]=1;//只有找到和当前的距离相等时才更新最短路径的数量 maxnum[s]=city[s];//记得初始化 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]; maxnum[j]=maxnum[t]+city[j]; pre[j]=t; cnt[j]=cnt[t]; }else if(dist[j]==dist[t]+g[t][j]) //找到和当前路径相等的路径,更新最短路径的数量 { cnt[j]+=cnt[t]; if(maxnum[j]<maxnum[t]+city[j]) { maxnum[j]=maxnum[t]+city[j]; pre[j]=t; } } } } } void print(int x) { if(x!=s) { print(pre[x]); cout<<" "<<x; }else cout<<x; } int main() { cin>>n>>m>>s>>d; memset(g,0x3f,sizeof g); for(int i=0;i<n;i++)scanf("%d",&city[i]); while(m--) { int a,b,c; cin>>a>>b>>c; g[a][b]=g[b][a]=min(g[a][b],c); } dijkstra(); cout<<cnt[d]<<" "<<maxnum[d]<<endl; print(d); }
最后附上一张图片,激励正在为梦想拼搏的你,fighting!
rr