L2-001 紧急救援 (25 分) 迪杰斯特拉算法求最短路径问题

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

分析
这是一道典型的最短路径问题，可以采用迪杰斯特拉算法解决，如果不懂此算法，请先了解后再看下面代码。此题在该算法的基础上，需要设一个数组来记录最短路径上每一个结点的前驱结点；一个数组记录最短路径前提下的最大救援队数量以及一个数组记录最短路径的个数。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
const int MaxSize=0x3f3f3f3f;
int n,m,s,k;
int a[505][505];   //用于存储两个地点之间路径的长度，不存在则设为无穷大 
int dist[505];		//dist[i]表示起点到i的最短路径长度 
int visited[505];	//visited[i]表示i点是否被访问过，在本代码中visited[i]初始为0,表示未访问，visited[i]=1，表示已访问 
int num[505];		// num[i]表示起点到i点存在的最短路径的个数 
int pre[505];		//pre[i]表示最短路径中上一个经过的城市的编号 
int wei[505];		//wei[i]表示i城市救援队数目 
int sumwei[505];	//sumwei[i]表示从起点到达i城市在最短路径情况下的最大救援队数量 

//该函数用于找出起点到各点最短的一条中点的编号 
int minPath(){
	int j=-1;
	for(int i=0;i<n;i++){
		if(j==-1&&visited[i]==0)     
			j=i;
		else if(j!=-1&&visited[i]==0&&dist[i]<dist[j]){
			j=i;
		}
	}
	return j;
}
void dijkstra(){
	memset(visited,0,sizeof(visited));
	memset(num,0,sizeof(num));
	memset(pre,-1,sizeof(pre));
	memset(sumwei,0,sizeof(sumwei));
	for(int i=0;i<n;i++){   //s起点到其他各点的路径长度 
		dist[i]=a[s][i];
		if(dist[i]<MaxSize){
			num[i]=1;		 
			pre[i]=s;		 
			sumwei[i]=wei[s]+wei[i];
		}
	}
	visited[s]=1;		//s点设置 为被访问过了 
	num[s]=1;
	pre[s]=-1;			//因为s点是起点，所以没有前驱结点 
	dist[s]=0;			 
	for(int i=0;i<n-1;i++){
		int j=minPath();
		if(j==-1)		
			break;
		visited[j]=1;		
		for(int k=0;k<n;k++){			//更新dist[] 
			if(visited[k]==0&&dist[k]>dist[j]+a[j][k]){
				dist[k]=dist[j]+a[j][k];
				num[k]=num[j];
				sumwei[k]=sumwei[j]+wei[k];
				pre[k]=j;
			}else if(visited[k]==0&&dist[k]==dist[j]+a[j][k]){  //如果 路径长度相等的情况下
				num[k]+=num[j];			// 因为起点到k点的路径长度等于s先到j点然后再从j点到k的长度，因此此时k点最短路径长度等于j点的最短路径加上原有的最短路径个数 
				if(sumwei[k]<sumwei[j]+wei[k]){      //因为题目说要召集尽可能多的救援队
					pre[k]=j;                      //更新前驱结点 
					sumwei[k]=sumwei[j]+wei[k];
				}
			}
		}
	}
	
}
int main(){
	cin>>n>>m>>s>>k;
	for(int i=0;i<n;i++){
		cin>>wei[i];
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			a[i][j]=MaxSize;
		}
	}
	for(int i=0;i<m;i++){
		int x,y,z;
		cin>>x>>y>>z;
		a[x][y]=z;
		a[y][x]=z;
	}
	dijkstra();
	cout<<num[k]<<" "<<sumwei[k]<<endl;
	int b[1000];
	int t=0;
	while(k!=-1){     //回溯终点到起点所经过的结点，存于b[]，最后再输出 
		b[t++]=k;
		k=pre[k];
	}
	for(int i=t-1;i>0;i--){
		cout<<b[i]<<" ";
	}
	cout<<b[0]<<endl;
	
}

运行截图：