Nyoj 城市平乱(图论)

描述

南将军统领着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

来源

《世界大学生程序设计竞赛高级教程·第一册》改编

上传者

张云聪

思想:  把终点作为起点,去找起点......get啦.....

代码:

#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <string>
#include <cstring>
#include <queue>
using namespace std;
#define INF 0x3f3f3f3f
#define MAX 1000000 

int n,m,p,q;
int num[111];
int mp[1111][1111];
int vis[1111];
int dis[1111];

void dijkstra(int s)
{
    for(int i=1; i<=m; i++)  
        dis[i] = mp[s][i];  

    dis[s]=0;
    vis[s]=1;

    for(int i=1; i<=m; i++){
        int min=INF;
        int k=0;
        for(int j=1; j<=m; j++){
            if(!vis[j] && min>dis[j]){
                min=dis[j];
                k=j;
            }
        }
        if(min==INF)
            break;
        vis[k]=1;
        for(int j=1; j<=m; j++){
            if(!vis[j] && dis[j]>min+mp[k][j])
                dis[j]=min+mp[k][j];
        }
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%d%d",&n,&m,&p,&q);
        memset(mp,INF,sizeof(mp));
        for(int i=0; i<n; i++){
            scanf("%d",&num[i]);
        }
        int a,b,c;
        for(int i=0; i<p; i++){
             scanf("%d %d %d", &a, &b, &c);  
            if(mp[a][b] > c)  
                mp[a][b] = mp[b][a] = c;  
        }
        dijkstra(q);
        int min = INF;
        for(int i=0; i<n; i++){
            if(dis[num[i]] < min)
                min=dis[num[i]];
        }
        printf("%d
",min);
        memset(vis,0,sizeof(vis));
    }
}