nyoj 115 Dijkstra

来源:互联网 发布:python 运行时间 编辑:程序博客网 时间:2024/04/30 04:27
描述

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

数据保证暴乱的城市是可达的。
输出
对于每组测试数据,输出第一支部队到达叛乱城市时的时间。每组输出占一行
样例输入
13 8 9 81 2 31 2 12 3 21 4 22 5 33 6 24 7 15 7 35 8 26 8 2 

样例输出 

           4

简单的最短路径问题,AC代码:

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
bool visited[1010];
int  Edge[1010][1010];
int  dis[100010];
int P,Q,N,M,temp,a[1010];
int Dijkstra(int,int);
int main()
{
int test;
cin>>test;
while(test--)
{
memset(Edge,100,sizeof(Edge));      //100对应的不是int的100,测试是一个很大的数。。 因为填充是以字节为单位的
cin>>N>>M>>P>>Q; 
int i,j,x,y,k,m=10000;
for(i=1;i<=N;++i)
{
cin>>a[i];
}
for(i=1;i<=P;++i)
{
scanf("%d%d%d",&x,&y,&k);
Edge[x][y]=Edge[y][x]=k;
}
for(j=1;j<=N;++j)
{
temp=Dijkstra(a[j],Q);
if(temp<m)
m=temp;
}
printf("%d\n",m);
}
//system("pause");
return 0;
}
//从s0开始的最短路径 
int Dijkstra(int s0,int q1)
{
memset(visited,false,sizeof(visited));
if(s0==q1) return 0;
    visited[s0]=true;
int i,j,sum=0;
memset(dis,100,sizeof(dis));
for(i=1;i<=M;++i)
{
dis[i]=Edge[s0][i];  //与so相邻点的相关边的权值
}
 for(int k=1;k<M;++k)       //n-1条边,n-1次循环 
 {
 int min=10000,q;
 for(i=1;i<=M;++i)      //找出当前权值最小值 
 if(dis[i]<min&&!visited[i]) 
    {
     min=dis[i];
 q=i;           //找到 
  }
  visited[q]=true;
  sum=sum+min;
  for(i=1;i<=M;++i)       //更新dis的值 
  if(dis[i]>min+Edge[q][i]&&!visited[i])
  dis[i]=min+Edge[q][i];

return dis[q1];
}
  

原创粉丝点击