NYOJ 115 城市平乱

城市平乱

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

分析：可以把该题看做求由各个部队所在的城市到暴乱城市的最短路径，然后求最小值。我一开始是用Floyd求出各个顶点的最短路径，然后求最小值，结果超时了。。。之后我把暴乱城市当起点，用Dijkstra算法求得从起点到各个点的最短路径，再求最小值，就AC了

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

using namespace std;

#define INF 0x3f3f3f3f

int n,m,p,q;
int a[1010][1010];//邻接矩阵
int b[110];
int s[1010];//标记
int dist[1010];//从起点到各个点的最短距离
int path[1010];//保存从起点到当前节点最短路径的上一个节点

void Dijkstra(int p)
{
    for(int i = 1;i <= m;i++)
    {
        dist[i] = a[p][i];
        s[i] = 0;
        if(i != p && dist[i] < INF)
            path[i] = p;
        else
            path[i] = -1;
    }
    s[p] = 1;dist[p] = 0;
    for(int i = 1;i <= m;i++)//从顶点p确定m - 1条最短路径
    {
        int minn = INF,u = p;
        for(int j = 1;j <= m;j++)//选择当前集合T中具有最短路径的顶点u
        {
            if(!s[j] && dist[j] < minn)
            {
                u = j;
                minn = dist[j];
            }
        }
        s[u] =1;//将顶点u加入到集合s，表示它的最短路径以求得
        for(int k = 1;k <= m;k++)//修改T集合中顶点dist和path数组元素值
        {
            if(!s[k] && a[u][k] < INF && dist[u] + a[u][k] < dist[k])
            {
                dist[k] = dist[u] + a[u][k];
                path[k] = u;
            }
        }
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d %d %d",&n,&m,&p,&q);
        for(int i = 0;i < n;i++)
        {
            scanf("%d",&b[i]);
        }
        memset(a,INF,sizeof(a));
        int u,v,w;
        for(int i = 0;i < p;i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            a[u][v] = w;
            a[v][u] = w;
        }
        memset(dist,INF,sizeof(dist));
        Dijkstra(q);//求从q点到各个点的最短路径
        //求得最小值
        int res = INF;
        for(int i = 0;i < n;i++)
        {
            res = min(res,dist[b[i]]);
        }
        printf("%d\n",res);
    }
    return 0;
}