南阳oj 题目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

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int mapp[1010][1010];//保存城市行军需要的时间
int n,m,x,y;
int jun[1010];//标记该城市是否有驻军
int dis[1010];//存储每个城市到目标城市的最短时间
int state[1010];//标记该城市是搜索过
void intit()//对城市间的距离进行初始化
{
    for(int i=1;i<=m;i++)
    {
        jun[i]=0;
        for(int j=1;j<=m;j++)
        {
            if(i==j)
                mapp[i][j]=0;
            else
                mapp[i][j]=0x3f3f3f3f;
        }
    }
}
int dij()
{
    for(int i=1;i<=m;i++)//将个点到目的地的距离赋值给数组dis[]
    {
        dis[i]=mapp[y][i];
        state[i]=0;
    }
    state[y]=1;
    int text=y;
    int num=1;
    while(num<m)
    {
        int maxx=0x3f3f3f3f;
        for(int i=1;i<=m;i++)
        {
            if(state[i]==0&&dis[i]<maxx)
            {
                text=i;
                maxx=dis[i];
            }
        }
        state[text]=1;
        num++;
        for(int i=1;i<=m;i++)
        {
            if(state[i]==0&&dis[i]>dis[text]+mapp[text][i])
                dis[i]=dis[text]+mapp[text][i];
        }
    }
    int sum=0x3f3f3f3f;
    for(int i=1;i<=m;i++)//找最短路径
    {
        if(sum>dis[i]&&jun[i]==1)
            sum=dis[i];
    }
    return sum;
}
int main()
{
    int k;
    scanf("%d",&k);
    int text,a,b,t;
    while(k--)
    {
        scanf("%d%d%d%d",&n,&m,&x,&y);
        intit();
        for(int i=1;i<=n;i++)//将存在的城市标记一下
        {
            scanf("%d",&text);
            jun[text]=1;
        }
        for(int i=1;i<=x;i++)
        {
            scanf("%d%d%d",&a,&b,&t);
            mapp[a][b]=mapp[b][a]=mapp[a][b]>t?t:mapp[a][b];//记录城市间的距离，比较的作用是为了防止两城市多次输入，且在不相同
        }
        printf("%d\n",dij());
    }
}