pta L2-001 紧急救援

L2-001 紧急救援

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

输入格式:

输入第一行给出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

大致思路：

采用迪杰斯特拉算法，以及借助pre数组记录前驱节点。

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 505;
int mm[maxn][maxn];  #记录地图，采用邻接矩阵
int pre[maxn];     #记录前驱节点保存最有路径
int dis[maxn];      #记录s到其余所有点最短距离
int w[maxn];        #记录s到其余所有点的最大点权
bool vis[maxn];    
int city[maxn];      #记录城市的点权
int num[maxn];		#s到个路径的最优路径数
void find(int s,int n){
	fill(vis,vis+maxn,false);
	fill(w,w+maxn,0);
	fill(dis,dis+maxn,0x3fffffff);
	fill(pre,pre+maxn,-1);
	fill(num,num+maxn,0);
	dis[s] = 0;
	w[s] = city[s];
	num[s] = 1;
	for(int i=0;i<n;i++){
		int u = -1,min = 0x3fffffff;
		for(int j=0;j<n;j++){
			if(dis[j]<min&&vis[j]==false){
				min = dis[j];
				u = j;
			}
		}
		if(u==-1) continue ;
		vis[u] = true;
		for(int v=0;v<n;v++){
			if(vis[v]==false&&mm[u][v]!=0x3fffffff){
				if(mm[u][v]+dis[u]<dis[v]){
					dis[v] = mm[u][v] + dis[u];
					w[v] = w[u] + city[v];
					num[v] = num[u];
					pre[v] = u;
				}
			else if(mm[u][v]+dis[u]==dis[v]){
					if(w[v]<w[u]+city[v]){
						w[v] = w[u] + city[v];
						pre[v] = u;
					}
					num[v] += num[u];
				}
			
			}
		}
	}
}
void print_path(int s,int x,int d){
	if(x==s){
		cout<<x<<" ";
		return ;
	}
	print_path(s,pre[x],d);
	if(x!=d)
		cout<<x<<" ";
	else
		cout<<x;
	
}
int main(void){
	for(int i=0;i<maxn;i++){
		for(int j=0;j<maxn;j++){
			mm[i][j] = 0x3fffffff;
		}
	}
	int n,m,s,d;
	cin>>n>>m>>s>>d;
	for(int i=0;i<n;i++){
		cin>>city[i];
	}
	for(int i=0;i<m;i++){
		int x,y,w;
		cin>>x>>y>>w;
		mm[x][y] = w;
		mm[y][x] = w;
	}
	find(s,n);
	cout<<num[d]<<" "<<w[d]<<endl;
	print_path(s,d,d);
	return 0;
}