NYOJ115 城市平乱

城市平乱

时间限制：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来标记到达某地点的最短距离，并且设置一个变量MIN来获得到达一个有军队的城市的当前最小距离，当遍历的dis>min时，就不在继续遍历。

看代码把，感觉自己讲不清楚。。。算法思想看书发现与bellman-ford算法思想一样。

感觉自己的代码在非最坏情况下的效率还不错，16MS。

代码如下：

//NY115 城市平乱


#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
using namespace std;

#define M 1010
const int MAX=0x7FFFFFFF;

int map[M][M];
int dis[M];        //从指定点到该店的距离，刷新至最小值
int army[M];       //军队地点
int m,q;

void init()
{
    memset(map,1,sizeof(map));           //虽说是按一个字节赋值，但反正超过100000，所以也无所谓了.
    memset(dis,1,sizeof(dis));
    memset(army,0,sizeof(army));        
}

int bellman_ford()
{
    queue<int> qu;
    int t,i,min=MAX;
    t=q;
    qu.push(t);
    while(!qu.empty())
    {
        t=qu.front();
        qu.pop();
        if(army[t]&&min>dis[t])
            min=dis[t];
        for(i=1;i<=m;i++)
        {
            if(map[t][i]&&(map[t][i]+dis[t]<dis[i]))
            {
                dis[i]=dis[t]+map[t][i];
                if(dis[i]<min)
                    qu.push(i);
            }
        }
    }
    return min;
}

int main(void)
{
    int t,n,p,i,temp,a,b,d;
//    freopen("in.txt","r",stdin);
    scanf("%d",&t);
    while(t--) 
    {
        scanf("%d%d%d%d",&n,&m,&p,&q);
        init();
        dis[q]=0;
        for(i=0;i<n;i++)
        {
            scanf("%d",&temp);
            army[temp]=1;
        }
        for(i=0;i<p;i++)
        {
            scanf("%d%d%d",&a,&b,&d);
            if(map[a][b]>d)
            {
                map[a][b]=d;
                map[b][a]=d;
            }
        }
        printf("%d
",bellman_ford());
    }
    return 0;
}