L2-001 紧急救援

·# L2-001 紧急救援 (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

思路分析：

1、很明显是单源最短路问题，选择dij算法

2、题目要求最短路的条数、召集的人数、路径，多开两个数组记录就好

解题代码：

#include<bits/stdc++.h>
using namespace std;

//最短路径条数、更多的救援队
int n,m,s,d;
const int N = 1001;
const int INF = 0x3f3f3f3f;
int vis[N];
int e[N][N];
int dist[N]; //记录最短距离
int person[N]; //记录最多人数
int per[N];
int cnt=0;
int pre[N];
int num[N];
void dij()
{
	dist[s]=0;
	person[s]=per[s];
	num[s]=1;
	for(int j=0;j<n;j++)
	{
		int index=-1,minn=INF;
		for(int i=0;i<n;i++)
		{
			if(dist[i]<minn&&!vis[i])
			{
				minn=dist[i];
				index=i;
			}
		}
		if(index==-1)
		{
			break;
		}
		vis[index]=1;
		for(int i=0;i<n;i++)
		{
			if(dist[i]>dist[index]+e[index][i]&&!vis[i])
			{
				dist[i]=dist[index]+e[index][i];
				person[i]=per[i]+person[index];
				pre[i]=index;
				num[i]=num[index];
			}
			else if(dist[i]==dist[index]+e[index][i]&&!vis[i])
			{
				num[i]=num[index]+num[i];//本来就能到达i的路径条数+通过index到达i的条数
				if(per[i]+person[index]>person[i]) //如果通过index到达该点所能得到的人数大于原来的人数，则更新路径
				{
					person[i]=per[i]+person[index];
					pre[i]=index; 
				}
				 
			}
		}
	}
}

void printPath(int d)
{
	if(d==s)
	{
		cout<<d;
		return ;
	}
	printPath(pre[d]);
	cout<<" "<<d;
	
}

int main()
{
	cin>>n>>m>>s>>d;
	for(int i=0;i<n;i++)
	{
		cin>>per[i];
	}
	fill(vis,vis+N,0);
	fill(e[0],e[0]+N*N,INF);
	fill(dist,dist+N,INF);

	while(m--)
	{
		int a,b,c;
		cin>>a>>b>>c;
		e[a][b]=e[b][a]=c;	
	} 
	dij();

	cout<<num[d]<<" ";
	cout<<person[d]<<endl;
	printPath(d);
}