L2-001. 紧急救援(迪杰斯特拉+dfs)

L2-001. 紧急救援

时间限制

200 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式：

输入第一行给出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>
#define inf 0x3f3f3f3f
using namespace std;
int mp[505][505],vis[505];
int team[505],path[505],count[505],dis[505];
int n,m,s,d;
void dfs(int dist,int teams,int id)
{
	if(id==s)
	return ;
	for(int i=0;i<n;i++)
	{
		if(dist-mp[i][id]==dis[i]&&teams-team[id]==count[i])
		{
			dfs(dis[i],count[i],i);
			cout<<i<<" ";
			break;
		}
	}
}
void djs()
{
	for(int i=0;i<n;i++)
	{
		dis[i]=mp[s][i];
		if(i!=s&&mp[s][i]!=inf)
		{
			path[i]=1;
			count[i]=team[s]+team[i];
		}
	}
	vis[s]=1;
	for(int i=0;i<n-1;i++)
	{
		int k,mincost=inf;
		for(int j=0;j<n;j++)
		{
			if(!vis[j]&&mincost>dis[j])
			{
				mincost=dis[j];
				k=j;
			}
		}
		vis[k]=1;
		for(int j=0;j<n;j++)
		{
			if(!vis[j]&&dis[j]>dis[k]+mp[k][j])
			{
				dis[j]=dis[k]+mp[k][j];
				path[j]=path[k];
				count[j]=count[k]+team[j];
			}
			else if(!vis[j]&&dis[j]==dis[k]+mp[k][j])
			{
				path[j]+=path[k];
				if(count[j]<count[k]+team[j])
				count[j]=count[k]+team[j];
			}
		}
	}
}
int main()
{
	memset(mp,inf,sizeof(mp));
	memset(vis,0,sizeof(vis));
	cin>>n>>m>>s>>d;
	for(int i=0;i<n;i++)
	cin>>team[i];
	for(int i=0;i<m;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		mp[u][v]=mp[v][u]=w;
	}
	djs();
	cout<<path[d]<<" "<<count[d]<<endl;
	cout<<s<<" ";
	dfs(dis[d],count[d],d);
	cout<<d;
}