NYOJ 题目115 城市平乱 （最短路径-迪杰斯特拉算法）

城市平乱

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 4

描述

南将军统领着N个部队，这N个部队分别驻扎在N个不同的城市。

他在用这N个部队维护着M个城市的治安，这M个城市分别编号从1到M。

现在，小工军师告诉南将军，第K号城市发生了暴乱，南将军从各个部队都派遣了一个分队沿最近路去往暴乱城市平乱。

现在已知在任意两个城市之间的路行军所需的时间，你作为南将军麾下最厉害的程序员，请你编写一个程序来告诉南将军第一个分队到达叛乱城市所需的时间。

注意，两个城市之间可能不只一条路。

输入

第一行输入一个整数T，表示测试数据的组数。(T<20)
每组测试数据的第一行是四个整数N,M,P,Q(1<=N<=100,N<=M<=1000,M-1<=P<=100000)其中N表示部队数，M表示城市数，P表示城市之间的路的条数,Q表示发生暴乱的城市编号。
随后的一行是N个整数，表示部队所在城市的编号。
再之后的P行，每行有三个正整数，a,b,t(1<=a,b<=M,1<=t<=100)，表示a,b之间的路如果行军需要用时为t

数据保证暴乱的城市是可达的。

输出

对于每组测试数据，输出第一支部队到达叛乱城市时的时间。每组输出占一行

样例输入

1
3 8 9 8
1 2 3
1 2 1
2 3 2
1 4 2
2 5 3
3 6 2
4 7 1
5 7 3
5 8 2
6 8 2 

样例输出

4

思路：用迪杰斯特拉算法+图的邻接矩阵，把叛乱城市当做源点，求到其他所有点的最短路径，然后在找出到有士兵的城市中最短路径最短的一个城市，输出路径长度。

代码如下：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int D[1005];//路径长度 
int final[1005];//final[i]为true代表已经找到到i的最短路径 
int e[1005][1005];//邻接矩阵 
bool army[1005];//值为true则该城市有士兵驻扎 
int n,m,p,q; 
void my_init(){
	memset(army,0,sizeof(army));
	for(int i=1;i<=m;i++)
	   for(int j=1;j<=m;j++)
	    e[i][j]=1000005;
	for(int i=1;i<=m;i++)
	  final[i]=false;   
}
int main(){
	int cases,i,j,x,a,b,t,v,w,min_path;
	cin>>cases;
	while(cases--){
		scanf("%d%d%d%d",&n,&m,&p,&q);
		my_init();//初始化 
		for(i=1;i<=n;i++){//有士兵的城市设为true 
			scanf("%d",&x);
			army[x]=true;
		}
		for(i=1;i<=p;i++){//构建邻接矩阵 
			scanf("%d%d%d",&a,&b,&t);
			e[a][b]=e[b][a]=t;
		}
		for(v=1;v<=m;v++)
		D[v]=e[q][v];//赋初值 
		D[q]=0;final[q]=true;//对源点赋值 
		for(i=1;i<m;i++){//找到其余m-1个点的最短路径，每次循环找到一个最短路径 
			int minn=1000005;
			for(w=1;w<=m;w++)
			   if(!final[w])
			      if(D[w]<minn){v=w;minn=D[w];}
		final[v]=true;	
		for(w=1;w<=m;w++){//找到一个最短路径后，更新其它路径的长度 
			if(!final[w]&&(minn+e[v][w])<D[w]){D[w]=minn+e[v][w];}
		}      
		}
		min_path=1000005;
		for(j=1;j<=m;j++){
			if(army[j]&&D[j]<min_path)min_path=D[j];//找到有士兵的，且最短的路径 
		}
		printf("%d\n",min_path);
	}
	return 0;
}