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

之前一直过不去测试点2，调了好长时间。事实证明这道题不是非得用Djkstra。A的那一刻快哭了，真的。

#include <bits/stdc++.h>
using namespace std;
int main(void){
	int n,m,s,d,dt,**p;
	cin>>n>>m>>s>>d;dt=d;
	int sav[n];									//救援队 
	int path[n];								//路径
	int ps[n]={0};								//路径条数
	int ss[n]={0};								//累计救援队数
	int ls[n];fill(ls,ls+n,0x7fffffff);			//累计距离 
	queue<int> f;								//访问队列
	stack<int> v;								//路径输出栈 
	p=(int**)malloc(sizeof(int*)*n);			//邻接矩阵
	for(int i=0;i<n;i++){
		p[i]=(int*)malloc(sizeof(int)*n);	
		for(int j=0;j<n;j++)p[i][j]=0;			//初始化 
		cin>>sav[i];
	}
	for(int i=0;i<m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		p[x][y]=p[y][x]=z;
	} 
	ss[s]=sav[s];								//起点累计救援队数
	path[s]=-1;									//-1作为哨兵
	ls[s]=0;									//起点累计距离 
	ps[s]=1;									//起点有一条路
	f.push(s);									//起点入队
	while(!f.empty()){
		int x=f.front();f.pop();				//出队
		for(int i=0;i<n;i++){
			if(p[x][i]){
				if(ls[i]>ls[x]+p[x][i]){		//距离更大 
					ls[i]=ls[x]+p[x][i];		//更新距离
					path[i]=x;					//更新路径 
					ps[i]=ps[x];				//路径条数
					ss[i]=ss[x]+sav[i];			//更新救援队 
					f.push(i);					//入队 
				}
				else if(ls[i]==ls[x]+p[x][i]){	//距离相等 
					ps[i]=0;					//把路径数清空重新计算 
					for(int j=0;j<n;j++)if(p[j][i]&&p[j][i]+ls[j]==ls[i])ps[i]+=ps[j];
					if(ss[i]<ss[x]+sav[i]){		//如果有更多的救援队方案 
						path[i]=x;				//更新路径 
						ss[i]=ss[x]+sav[i];		//更新救援队 
						f.push(i);				//重新入队 
					}
				} 
			}
		} 
	}
	do{
		v.push(dt);								//入栈 
		dt=path[dt];							//找上一个节点 
	}while(dt!=-1);								//如果上一个节点不为-1，继续 
	cout<<ps[d]<<' '<<ss[d]<<endl; 
	while(!(v.empty())){
		cout<<v.top();v.pop();
		if(!(v.empty()))cout<<' ';
	}
	return 0;
}