NYOJ 115 城市平乱（图论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算法的源点，即求暴乱城市到各个城市的最短路

#include<string.h>
#include<stdio.h>
#define INF 9999999
using namespace std;
///迪杰斯特拉最短路
int Map[1005][1005];///存储地图数据
int dis[1005];///存储距离
int vis[1005];///已走过的点标记为1，初始化为0


void Dijsktra(int s, int N) ///从原点s到其他N-1个点的距离
{
    for(int i=1; i<=N; i++) ///初始化距离数组
        dis[i]=Map[s][i];
    vis[s]=1;///s到各点距离，所以s已经被访问
    int minn,u;///辅助变量
    for(int i=1; i<=N-1; i++) ///迪杰斯特拉核心语句
    {
        minn=INF;
        for(int j=1; j<=N; j++)
        {
            if(vis[j]==0&&dis[j]<minn)
            {
                minn=dis[j];
                u=j;
            }
        }
        vis[u]=1;///标记为已访问
        for(int v=1; v<=N; v++)
        {
            if(Map[u][v]<INF)
            {
                if(dis[v]>dis[u]+Map[u][v])
                    dis[v]=dis[u]+Map[u][v];///松弛完成
            }
        }
    }
}

int main()
{
    int K;
    scanf("%d",&K);
    while(K--)
    {
        int n, m, p, q;
        int army[105];
        scanf("%d%d%d%d", &n, &m, &p, &q);
        for(int i=0; i<n; i++)
            scanf("%d", &army[i]);
        for(int i=0; i<1005; i++) ///初始化
        {
            for(int j=0; j<1005; j++)
            {
                if(i==j) Map[i][j] = 0;
                else Map[i][j] = INF;
            }
        }
        for(int i=0; i<p; i++)
        {
            int a, b, t;
            scanf("%d%d%d", &a, &b, &t);
            if(t<Map[a][b]) {
                Map[a][b] = t;
                Map[b][a] = t;
            }
        }
        memset(vis,0,sizeof(vis));///已访问的标记为1
        Dijsktra(q, m);
        int minx = INF;
        for(int i=0; i<n; i++)
        {
            if(dis[army[i]]<minx) minx = dis[army[i]];
        }
        printf("%d\n", minx);
    }
    return 0;
}