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

/* 第一次用优先队列做了好久没做出来，然后又去把dijstra算法好好看了一遍，然后结合着自己看别人写的，最终还是做了出来（自己写的还是太少了，劝告大一的学弟学妹一定要好好学习啊），大家加油啊*/

在这里插入代码片#include<iostream>
#include<queue>
#include<string.h>
using namespace std;

const int inf = 1e6;      
int city[505][505];     //用来标记一个城市到另一个城市的距离	 
int pre[505];           //最短路径中城市的前驱 
bool sign[505];         //标记一个城市是否在最短路径中 
int arm[505];           //该城市的军队数目 
int sum[505];           //经过最短路径中当前城市的军队总数 
int dis[505];           //该城市到起点的最短距离 
int art[505] = {0};     //最短路径的条数 
struct node{
	int n;         //城市 
	int dis;       //该城市距离起点的最短距离 
	node(int _n,int _dis)
	{
		n = _n;
		dis = _dis;
	}
	bool operator < (const node &a)const   //优先队列的优先性，距离起点近的优先级别高 
	{
		return a.dis < dis;
	}
};
int n,m,be,en;

void dijstra(int s)
{
	memset(dis,inf,sizeof(dis));
	dis[s] = 0;                        //起点到起点的距离为0	 
	sum[s] = arm[s];				   //起点到起点军队总数 
	art[s] = 1;							//起点到起点的最短路径条数为  1 
	priority_queue<node>q;
	q.push(node(s,dis[s]));
	while(!q.empty())
	{
		node u = q.top();
		q.pop();
		if(sign[u.n])
		continue;
		sign[u.n] = true;
		for(int i=0;i<505;i++)
		{
			if(city[u.n][i]!=inf)
			{
				if(sign[i])
				continue;
				if(dis[i] > dis[u.n] + city[u.n][i])		//距离相比较短 
				{
					dis[i] = dis[u.n] + city[u.n][i];
					sum[i] = arm[i] + sum[u.n];
					pre[i] = u.n;
					q.push(node(i,dis[i]));
					art[i] = art[u.n];                     //更新最短路径条数 
				}
				else if(dis[i] == dis[u.n]+city[u.n][i])	//距离相等，但是军队总数较多 
				{
					if(sum[i] < sum[u.n] + arm[i])
					{
						sum[i] = arm[i] + sum[u.n];
						pre[i] = u.n;
						q.push(node(i,dis[i]));
					}
					art[i]+=art[u.n];                        //更新最短路径条数 
				}
			}
		}
	}
} 
void print(int be,int en)                                  //遍历起点到终点经过的城市 
{
	if(be == en)
	{
		printf("%d",en);
		return ;
	}
	else
	{
		print(be,pre[en]);
		printf(" %d",en);
	}
}

int main()
{
	scanf("%d %d %d %d",&n,&m,&be,&en);
	for(int i=0;i<n;i++)
	{
		int a;
		scanf("%d",&a);
		arm[i] = a;
	}
	memset(pre,-1,sizeof(pre));
	memset(sum,0,sizeof(sum));
	memset(sign,false,sizeof(sign));
	for(int i=0;i<505;i++)
	for(int j=0;j<505;j++)
	city[i][j] = inf;
	for(int i=0;i<m;i++)
	{
		int a,b,w;
		scanf("%d %d %d",&a,&b,&w);
		city[a][b] = w;
		city[b][a] = w;
	}
	dijstra(be);
	cout<<art[en]<<" "<<sum[en]<<endl;
	print(be,en);
}