7-37 城市间紧急救援 (25 分)
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出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<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define Inf 0x3f3f3f3f
const int maxn=1005;
int g[maxn][maxn];
int cnt[maxn],vis[maxn],cf[maxn],ccf[maxn],pre[maxn];
//最短路径条数,各城市是否经过,各城市救援队数量,到达该城市时所召集的救援队数量,到达该城市前经过的城市编号
int n,m,s,d;
//初始化
void init()
{
cin>>n>>m>>s>>d;
for(int i=0;i<n;i++)
{
cin>>cf[i];
ccf[i]=cf[i];
cnt[i]=1;
}
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
if(i!=j)
g[i][j]=g[j][i]=Inf;
int a,b,c;
for(int i=0;i<m;i++)
{
cin>>a>>b>>c;
g[a][b]=g[b][a]=c;
}
}
void Dijkstra()
{
cnt[s]=1;
vis[s]=1;
for(int k=0;k<n;k++)
{
int minn=Inf,u=-1;
for(int i=0;i<n;i++)
{
if(!vis[i]&&g[s][i]<minn)
{
minn=g[s][i];
u=i;
}
}
if(u==-1) break;//与其他城市不连通,推出循环
else vis[u]=1;//到达下一个城市
for(int j=0;j<n;j++)
{
if(!vis[j]&&g[s][j]>g[s][u]+g[u][j])
{
g[s][j]=g[s][u]+g[u][j];//最短路径更新
cnt[j]=cnt[u];//拷贝到达上一个城市时的最短路径条数
pre[j]=u;//记录上一个城市的编号
ccf[j]=ccf[u]+cf[j];//到达某城市召集的全部救援队数量
}
else if(!vis[j]&&g[s][j]==g[s][u]+g[u][j])//发现其他最短路径
{
cnt[j]=cnt[u]+cnt[j];//更新到达当前城市时的最短路径条数
if(ccf[j]<ccf[u]+cf[j]) //最多救援队数量更新
{
pre[j]=u;//记录上一个城市的编号
ccf[j]=ccf[u]+cf[j];//更新救援队数量
}
}
}
}
}
int main()
{
init();
Dijkstra();
cout<<cnt[d]<<" "<<ccf[d]+cf[s]<<endl;
int k=0,t=d;
int road[maxn];
while(pre[t]!=0)
{
road[k++]=pre[t];
t=pre[t];
}
cout<<s<<" ";
for(int i=k-1;i>=0;i--)
cout<<road[i]<<" ";
cout<<d<<endl;
return 0;
}