nyoj 115 城市平乱

最新推荐文章于 2018-05-06 16:59:42 发布

原创最新推荐文章于 2018-05-06 16:59:42 发布 · 511 阅读

0 ·

CC 4.0 BY-SA版权

图论专栏收录该内容

12 篇文章

订阅专栏

城市平乱

时间限制：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

数据保证暴乱的城市是可达的。

输出

对于每组测试数据，输出第一支部队到达叛乱城市时的时间。每组输出占一行

样例输入

样例输出

用Djstra算法写，依次找到起始点与终点的最小值进行比较，水过。。

但时间上用时很大，还可以优化（这题也可以用搜索写）

#include "iostream"
#include "cstdio"
#include "cstring"
using namespace std;
const int MaxN = 1005;
const int INF = 0x3f3f3f3f;
int map[MaxN][MaxN]; //保存i点到j的点权值
int low[MaxN] ;  //用于存储到各点最短路径的权值和
int Path[MaxN];//用于存储最短路径下标的数组，此题没用到
int N,M;
int min(int a,int b){return a<b?a:b;}
int ShortPath_Dijkstar(int V0,int V1)
{
	int i ,j , k , pos , min ;
	int vis[1001];
    memset(vis,0,sizeof(vis));
//    memset(Path,0,sizeof(Path));
    //初始化数据
	for(i = 1 ; i <= N ; i++)
        low[i]=map[V0][i];
	low[V0] = 0;//v0到v0的路径为0
    vis[V0] = 1;//vis[i]=1表示已求得顶点V0到Vi的最短路径
   //开始主循环，每次求得v0到某个j顶点的最短路径
	for(i = 1 ; i < N ; i++)//i只表示次数，第一个点已确定，所以从第二个点开始，循环N-2次
	{
	    min = INF;
	    for(j = 1 ;  j <= N ; j++)//j表示到第j个点
	    if(!vis[j] && min > low[j])
	    {
	        min = low[j];
	        pos = j;
	    }
	    vis[pos] = 1;//将目前找到最近的顶点标记
	    //修正当前最短路径及距离
	    for(k = 1 ; k <= N ; k++ )
	    {//如果经过V顶点的路径比现在该条路径长度短的话，更新
	        if(!vis[k] && (min + map[pos][k]) < low[k])
	        {
	            low[k] = min + map[pos][k];
//	            Path[k] = pos ;
	        }
	    }
	}
//	for(k = 1 ; i <= N ;i++)
//	printf("%d ",low[i]);
//	printf("\n");
	return low[V1];
}

int main(){
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int k,q,K[110],i,j,a,b,c,ans;
        memset(K,0,sizeof(K));
        memset(map,INF,sizeof(map));

	    scanf("%d%d%d%d",&k,&N,&M,&q);
	    for(i = 1 ; i <= k ;i++)
            scanf("%d",&K[i]);//保存驻扎军队城市

	    for( i = 1 ; i <= M ; i++){
            scanf("%d%d%d",&a,&b,&c);
            map[a][b]=map[b][a]=c;
	    }
	    ans=INF;
        for(i = 1; i <= k ; i++)
	    {
	        ans=min(ans,ShortPath_Dijkstar(K[i],q));
	    }
	    printf("%d\n",ans);
	}
	return 0;
}