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

本题为单源最短路问题，可以学习dis的基本用法。

#include<iostream>
#include<stdio.h>
#include<cstring>
#define INF 0x3f3f3f3f
using namespace std;
int N,M,P,Q;//军队数，城市数，路的条数，发生暴乱的的城市编号
int jun[105];
int tu[1005][1005];//存路径
int dis[1005];//存每个城市到Q城市的最短路
int bj[1005];//标记该点是否访问过
int dij()
{
    for(int i=1;i<=M;i++)
    {
        dis[i]=tu[Q][i];//将各i点到Q点的路径存入dis数组
        bj[i]=0;//没访问的点都标记为0
    }
    bj[Q]=1;//访问的点标记为1
    int flag=Q;//下一个起始点
    int cut=1;//访问点的的个数
    while(cut<M)
    {
        int Min=INF;
        //把第一支军队到大Q点的最短时间初始化为无穷大
        for(int i=1;i<=M;i++)
        {//找出到Q点最近的点
            if(bj[i]==0&&dis[i]<Min)
            {//如果该点没有被访问并且较短
                Min=dis[i];//更新找到的最短路
                flag=i;//记下到Q点最近点的编号，作为下一个起始点
            }
        }
        bj[flag]=1;//标记找到的点，证明该点被访问过
        cut++;//城市个数加一
        for(int i=1;i<=M;i++)//用找到的最短路去更新别的路
        {
            if(bj[i]==0&&dis[i]>Min+tu[flag][i])
                dis[i]=Min+tu[flag][i];
        }
    }//然后找到下一个到Q点的最短路，然后用他更新别的路
    //使dis数组存的都是各点到Q点的最短路
       int Min=INF;//再把最短路设为无穷大
       for(int i=1;i<=M;i++)
       {
           if(jun[i]==1&&dis[i]<Min)
            //找出有军队并且到Q点最近的城市
               Min=dis[i];
        }
    return Min;
}
int main()
{
    int a,b,d,e,f;
    scanf("%d",&a);
    while(a--)
    {
        memset(jun,0,sizeof(jun));
        scanf("%d %d %d %d",&N,&M,&P,&Q);
        while(N--)
        {
            scanf("%d",&b);
            jun[b]=1;//将有军队的城市标记一下
        }
        for(int i=1;i<=M;i++)
        {
            for(int j=1;j<=M;j++)
            {
                tu[i][j]=INF;//初始化任意两条路之间的距离为无穷大
            }
        }
        while(P--)
        {
            scanf("%d %d %d",&d,&e,&f);
            if(tu[d][e]>f)//将路径存入
                tu[d][e]=tu[e][d]=f;//防止有重复路径出现
        }
        printf("%d\n",dij());
    }
    return 0;
}