城市平乱_NYOJ_115(dijkstra).java

城市平乱

时间限制： 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

来源

《世界大学生程序设计竞赛高级教程·第一册》改编

//AC
#include<stdio.h>
#include<string.h>
#define MAX 1000000
int city[101],map[1010][1010],d[1010],vis[1010],m;
void dijkstra(int start)
{
	for(int i=1;i<=m;i++){
		d[i]=map[start][i];
	}
	d[start]=0;
	for(int i=1;i<=m;i++){
		int min = MAX;
		int x=0;
		for(int j=1;j<=m;j++){
			if(!vis[j]&&min>=d[j]){
				min=d[j];
				x=j;
			}
		}
		vis[x]=1;
		for(int j=1;j<=m;j++){
			if(d[j]>min+map[x][j]){
				d[j]=min+map[x][j];
			}
		}
	}		
}
int main()
{
	int N,i,j;
	scanf("%d",&N);
	while(N--)
	{
		int n,p,q;
		scanf("%d%d%d%d",&n,&m,&p,&q);
		for(i=0;i<n;i++)
			scanf("%d",&city[i]);
		memset(map,MAX,sizeof(map));
		memset(vis,0,sizeof(vis));
		for(i=0;i<p;i++){
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			if(map[a][b]>c)
				map[a][b]=map[b][a]=c;
		}
		int temp=MAX;
		dijkstra(q);
		for(i=0;i<n;i++)
			if(temp>d[city[i]])
				temp=d[city[i]];
		printf("%d\n",temp);
	}
	return 0;
}

import java.util.Arrays;
import java.util.Scanner;
public class Main{//RuntimeError	 --	 --	java	08-20 21:45:11
	static int city[],MAX=1000000,m;
	static int map[][],d[];
	static boolean vis[];
	public static void main(String[] args) {
		Scanner input=new Scanner(System.in);
		int N=input.nextInt();
		while(N-->0){
			int n=input.nextInt();
			m=input.nextInt();
			int p=input.nextInt();
			int q=input.nextInt();
			city=new int[n+1];
			vis=new boolean[m+5];
			map=new int[m+5][m+5];
			d=new int[m+5];
			for(int i=0;i<n;i++)
				city[i]=input.nextInt();
			for(int i=1;i<=m;i++){
				Arrays.fill(map[i],MAX);
			}
			for(int i=0;i<p;i++){
				int a=input.nextInt();
				int b=input.nextInt();
				int c=input.nextInt();
				if(map[a][b]>c)
					map[a][b]=map[b][a]=c;
			}
			int temp=MAX;
			dijkstra(q);
			for(int i=0;i<n;i++){
				if(temp>d[city[i]])
					temp=d[city[i]];
			}
			System.out.println(temp);
		}
	}
	private static void dijkstra(int start) {
		for(int i=1;i<=m;i++){
			d[i]=map[start][i];
		}
		d[start]=0;
		for(int i=1;i<=m;i++){
			int min = MAX;
			int x=0;
			for(int j=1;j<=m;j++){
				if(!vis[j]&&min>=d[j]){
					min=d[j];
					x=j;
				}
			}
			vis[x]=true;
			for(int j=1;j<=m;j++){
				if(d[j]>min+map[x][j]){
					d[j]=min+map[x][j];
				}
			}
		}
	}
}

java至今未找到为啥运行时间错误，有发现的朋友请告知，非常感谢！