7-1 城市间紧急救援 (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 <set>
#include <map>
#include <cmath>
#include <stack>
#include <queue>
#include <vector>
#include <string>
#include <cstdio>
#include <cstring>
#include <sstream>
#include <iomanip>
#include <iostream>
#include <algorithm>
#define clr(str,x) memset(str,x,sizeof(str))
#define FRER() freopen("in.txt","r",stdin);
#define FREW() freopen("out.txt","w",stdout);
#define INF 0x3f3f3f3f
#define maxn 1000
typedef long long int ll;
using namespace std;
int n,m,S,D;
int len[maxn][maxn];//两点之间的距离,没有路记为INF
int sum[maxn];//到i点救援队的最大数量
int pre[maxn];//记录第i个点的前一个点
int d[maxn];//d[i]从S点到i点的最短距离
int man[maxn];//i点救援队数量
int Min,minid;//最小值、最小值的下标
bool vis[maxn];//记录这个点是否经过过了
int coun[maxn];//S到i点有多少条最短路
void Dijkstra()
{
clr(vis,false);
for(int i=0; i<n; i++)
{
d[i]=len[S][i];
if(len[S][i]!=0&&len[S][i]!=INF)
sum[i]+=sum[S];//如果S与i之间有路,i的救援队总数更新
pre[i]=-1;
coun[i]=1;
}
vis[S]=1;
d[S]=0;
for(int k=0; k<n; k++)
{
Min=INF,minid=-1;
for(int j=0; j<n; j++)
{
if(!vis[j]&&d[j]<Min)
{
Min=d[j];
minid=j;
}
}
if(minid==-1)
break;
vis[minid]=1;
for(int j=0; j<n; j++)
{
if(d[j]>Min+len[minid][j]&&!vis[j])
{
coun[j]=coun[minid];//更新S到第i个点的最短路数量
pre[j] = minid;
d[j]=Min+len[minid][j];
sum[j] = man[j]+sum[minid];
}
else if(d[j]==Min+len[minid][j]&&!vis[j])
{
coun[j]=coun[j]+coun[minid];
if(sum[j]<sum[minid]+man[j])//如果救援队数有更大的,就更新
{
sum[j]=sum[minid]+man[j];
pre[j]=minid;
}
}
}
}
}
int main()
{
//FRER()
//FREW()
int p1,p2,lens;
scanf("%d%d%d%d",&n,&m,&S,&D);
for(int i=0; i<n; i++)
{
for(int j=0; j<n; j++)
{
if(i==j)
len[i][j]=0;
else
len[i][j]=INF;
}
}
for(int i=0; i<n; i++)
{
scanf("%d",&man[i]);
sum[i]=man[i];
}
for(int i=0; i<m; i++)
{
scanf("%d%d%d",&p1,&p2,&lens);
len[p1][p2]=lens;
len[p2][p1]=lens;
}
Dijkstra();
printf("%d %d\n",coun[D],sum[D]);
int road[maxn];
memset(road,0,sizeof(road));
int pos=0,t=D;//从终点开始往前找
road[pos++]=D;
while(pre[t]!=-1)
{
road[pos++]=pre[t];
t=pre[t];
}
road[pos++]=S;
for(int i=pos-1; i>=0; i--)
{
if(i==pos-1)
printf("%d",road[i]);
else
printf(" %d",road[i]);
}
printf("\n");
return 0;
}