南阳理工OJ115（最短路径）

城市平乱

描述

南将军统领着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

View Code

1 #include<iostream>
2 #include<algorithm>
3 using namespace std;
4 #define Max 1001
5 #define MAX 1000000
6 int map[Max][Max],sign[Max],m;
7 bool cmp(int a, int b)
8 {
9 return a<b;
10 }
11 void ShortestPath(int k)
12 {
13 int i,j,min,u;
14 sign[k]=1;//初始化顶点k 属于S 集
15 //开始主循环，每次求得k 到某个u 顶点的最短路径，并加u 到S集
16 for(i=1;i<m;i++) //其余m-1 个顶点
17 {
18 min=MAX;
19 for(j=1;j<=m;j++)
20 if(sign[j]!=1&&min>map[k][j])//j顶点在V-S中
21 {
22 min=map[k][j]; //j顶点离k 顶点更近
23 u=j;
24 }
25 sign[u]=1; //离k 顶点最近的u 加入S集
26 for(j=1;j<=m;j++) //更新当前最短路径及距离
27 if(!sign[j]&&(min+map[u][j]<map[k][j]))
28 map[k][j]=min+map[u][j];
29 }
30 }
31 int main()
32 {
33 int T,n,p,q,an[101],i,j,a,b,t;
34 cin>>T;
35 while(T--)
36 {
37 cin>>n>>m>>p>>q;
38 for(i=0;i<n;i++) cin>>an[i];
39 for(i=1;i<=m;i++) //初始化
40 {
41 sign[i]=0;
42 for(j=1;j<=m;j++)
43 map[i][j]=MAX;
44 }
45 while(p--)
46 {
47 cin>>a>>b>>t;
48 if(map[a][b]>t)
49 map[a][b]=map[b][a]=t;
50 }
51 ShortestPath(q);
52 for(i=0;i<n;i++)
53 an[i]=map[q][an[i]]; //将驻军城市和叛乱城市的最短距离储存在数组an 中
54 sort(an,an+n,cmp);//快排
55 cout<<an[0]<<endl;
56 }
57 return 0;
58 }