NYOJ 115 城市平乱

描述

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

一开始想走捷径，用flord,很果断的给我一个TLE，其实反过来考更容易些，求暴乱城市到各个军队的距离，这样一个dijkstra就可以搞定。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 0x3f3f3f3f
using namespace std;
int map[1010][1010];
int a[110],dis[1010],book[1010];
int main()
{
    int i,j,k,n,m,p,q,t;
    scanf("%d",&t);
    while(t--)
    {
        memset(a,0,sizeof(a));
        memset(book,0,sizeof(book));
        scanf("%d%d%d%d",&n,&m,&p,&q);
        for(i=1; i<=m; i++)
            for(j=1; j<=m; j++)
                if(i==j) map[i][j]=0;
                else map[i][j]=N;
        for(i=0; i<n; i++)
            scanf("%d",&a[i]);
        int u,v,w;
        for(i=0; i<p; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            if(map[u][v]>w)
                map[u][v]=map[v][u]=w;
        }
        for(i=1; i<=m; i++)
            dis[i]=map[q][i];
        book[q]=1;
        for(i=1; i<m; i++)
        {
            int temp=N,pos=q;
            for(j=1; j<=m; j++)
            {
                if(book[j]==0&&dis[j]<temp)
                {
                    temp=dis[j];
                    pos=j;
                }
            }
            book[pos]=1;
            for(j=1; j<=m; j++)
            {
                if(dis[j]>dis[pos]+map[pos][j])
                    dis[j]=dis[pos]+map[pos][j];
            }
        }
        int min=N;
        for(i=0; i<n; i++)
            if(min>dis[a[i]])
                min=dis[a[i]];
        printf("%d\n",min);
    }
    return 0;
}