Dijkstra--城市平乱

城市平乱

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

来源

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

思路：题意就是求所有部队中能到叛乱城市所需的最短时间。逆向思维，即求叛乱城市到部队的最短时间，即从叛乱城市开始，到所有部队的最短路

问题。因为dijkstra算法求出的dis数组即为从始点开始到其他各点的最短路径，建好图后，dijkstra求出有部队的城市中dis数组的最小值即可。

#include<stdio.h>
#include<iostream>
#include<vector>
#include<string.h>

using namespace std;

typedef struct{
	int v;
	int weight;
}node;

vector <node>g[1001];
int n, m, p, dis[1001];

void init();
void read();
void dijkstra(int v);
int main(){
	int T, q, ans, i, j, army[101];
	scanf("%d", &T);
	while(T--){
		scanf("%d %d %d %d", &n, &m, &p, &q);
		for(i = 0; i < n; i++)
			scanf("%d", &army[i]);
		init();
		read();
		dijkstra(q);
		ans = 65535;
		for(i = 0; i < n; i++){
			if(ans > dis[army[i]]){
				ans = dis[army[i]];
			}
		}
		printf("%d\n", ans);
	}
	return 0;
}

void init(){
	int i;
	for(i = 1; i <= m; i++)
		g[i].clear();
}

void read(){
	int i, j, k, w;
	node e;
	for(i = 0; i < p; i++){
		scanf("%d %d %d", &j, &k, &w);
		e.v = k;
		e.weight = w;
		g[j].push_back(e);
		e.v = j;
		e.weight = w;
		g[k].push_back(e);
	}
}

void dijkstra(int v){
	int i, cur, weight, dist, intree[1001];
	for(i = 1; i <= m; i++){
		dis[i] = 65535;
		intree[i] = 0;
	}
	dis[v] = 0;
	
	while(intree[v] == 0){
		intree[v] = 1;
		for(i = 0; i < g[v].size(); i++){
			cur = g[v][i].v;
			weight = g[v][i].weight;
			if(dis[cur] > dis[v] + weight){
				dis[cur] = dis[v] + weight;
			}
		}

		dist = 65535;
		for(i = 1; i <= m; i++){
			if(intree[i] == 0 && dist > dis[i]){
				dist = dis[i];
				v = i;
			}
		}
	}
}