NYOJ 115 城市平乱 （Dijkatra）

城市平乱

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

来源《世界大学生程序设计竞赛高级教程·第一册》改编
Dijakatra:
AC代码：
#include <stdio.h>
#include <string.h>
#define INF 0x3f3f3f3f
int G[1005][1005];
int vis[1005];
int cost[1005]; 
int m,n,t,e;//部队数  城市数  路  暴乱城市 
void Dijkatra(int s)
{
int d,min;
vis[s]=1;
cost[s]=0;
for(int i=0;i<=n;i++)
cost[i]=G[s][i];
for(int i=1;i<=n;i++)
{   
   min=INF;
for(int j=0;j<=n;j++)
{
if(cost[j]<min&&vis[j]==0)
{
min=cost[j];
d=j;
}
}
vis[d]=1;
for(int j=0;j<=n;j++)
if(cost[j]>cost[d]+G[d][j]&&vis[j]==0)
cost[j]=cost[d]+G[d][j];
}
}
int main()
{
int k,a,b,c;
scanf("%d",&k);
while(k--)
{
scanf("%d %d %d %d",&m,&n,&t,&e);
memset(G,INF,sizeof(G));
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++)
G[i][i]=0;
int M;//驻扎的城市 
for(int i=0;i<m;i++)
{
scanf("%d",&M);
G[0][M]=0;
G[M][0]=0;
}
//这道题是m个源点,求从哪个源点到暴乱城市用时最短 
//在图里添加一个起点0，从0到各部队驻扎的城市的时间都是0
//这样就将m源点转化成了单源点，直接求从0出发到暴乱城市的最短时间即可
while(t--)
{
scanf("%d %d %d",&a,&b,&c);
if(c<G[a][b])//防止输入错误，保留最小的时间 
{
G[a][b]=c;
   G[b][a]=c;
}
}
Dijkatra(0);
printf("%d\n",cost[e]);
}
return 0;

} 

Floyd超时了。。。也贴出来吧
#include <stdio.h>
#include <string.h>
#define INF 0x3f3f3f3f
int G[1005][1005];
int m,n,t,s;//部队数  城市数  路  暴乱城市 
void floyd()
{
for(int k=1;k<=n;k++)
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(G[i][k]+G[k][j]<G[i][j])
G[i][j]=G[i][k]+G[k][j];
}
}
}
}
int main()
{
int k,a,b,c;
scanf("%d",&k);
while(k--)
{
scanf("%d %d %d %d",&m,&n,&t,&s);
memset(G,INF,sizeof(G));
for(int i=1;i<=n;i++)
G[i][i]=0;
int d[105];
for(int i=0;i<m;i++)
scanf("%d",&d[i]);
while(t--)
{
scanf("%d %d %d",&a,&b,&c);
if(c<G[a][b])//防止输入错误，保留最小的时间 
{
G[a][b]=c;
   G[b][a]=c;
}
}
floyd();
int min=INF;
for(int i=0;i<m;i++)
if(G[d[i]][s]<min) min=G[d[i]][s];//比较每个部队到暴乱城市所用的时间，选出最少的 
printf("%d\n",min);
}
return 0;
}