南阳oj里的一道题:115
刚开始的弗洛伊德算法(超时的):
#include<stdio.h>
int dist[101][1001];
int main()
{
int T,i;scanf("%d",&T);
while(T--)
{
int N,M,P,Q;int ni[101];
scanf("%d%d%d%d",&N,&M,&P,&Q);
for(i=0;i<N;i++)
scanf("%d",&ni[i]);
for(i=1;i<=M;i++)
for(int j=1;j<=M;j++)
dist[i][j]=101;
while(P--){
int a,b,t;
scanf("%d%d%d",&a,&b,&t);
dist[a][b]=t;
}
for(int k=1;k<=M;k++)
for(i=1;i<=M;i++)
for(int j=i+1;j<=M;j++)
if(dist[i][k]+dist[k][j]<dist[i][j]){
dist[i][j]=dist[i][k]+dist[k][j];
}
int min=dist[ni[0]][Q];
for(i=1;i<N;i++){
if(dist[ni[i]][Q]<min)
min=dist[ni[i]][Q];
}
printf("%d\n",min);
}
return 0;
}
看了别人总结的迪杰斯特拉算法:
http://blog.csdn.net/whjkm/article/details/38459733模板如下:
<pre name="code" class="cpp">void dijkstra(int start)//从一个给定的点开始 low[i]为start到i的路径长
{
int i,j,min,pos;
memset(visit,0,sizeof(visit));//初始化visit数组
for(i=1;i<=m;i++)
{
low[i]=map[start][i];
}
low[start]=0;//自身的权值为0
visit[start]=1;
for(i=1;i<m;i++)
{
min=Max;
pos=0;
for(j=1;j<=m;j++)
{
if(!visit[j] && min>low[j])//比较最小值,更新min值
{
min=low[j];
pos=j;
}
}
visit[pos]=1;
for(j=1;j<=m;j++)
{
if(low[j]>min+map[pos][j])//更新low数组
low[j]=min+map[pos][j];
}
}
}
通过代码如下:
#include<stdio.h>
#include<string.h>
#define inf 0x3f3f3f3f
#define Vex 1001
int map[Vex][Vex];int low[Vex];int M;int visit[Vex];
void dijstral(int start)
{
int i,j,min,pos;
for(i=1;i<=M;i++)
low[i]=map[start][i];
memset(visit,0,sizeof(visit));
low[start]=0;//自身权值为0
visit[start]=1;
for(i=1;i<=M;i++)
{
min=inf;pos=0;
for(j=1;j<=M;j++)
{
if(!visit[j]&&low[j]<min){
min=low[j];
pos=j;
}
}
visit[pos]=1;
for(j=1;j<=M;j++){
if(min+map[pos][j]<low[j]){
low[j]=min+map[pos][j];
}
}
}
}
int main()
{
int T;int N,P,Q;int num[101];
scanf("%d",&T);
while(T--){
memset(map,inf,sizeof(map));
scanf("%d%d%d%d",&N,&M,&P,&Q);int i;
for(i=0;i<N;i++)
scanf("%d",&num[i]);
while(P--)
{
int a,b,t;
scanf("%d%d%d",&a,&b,&t);
if(map[a][b]>t)
map[a][b]=map[b][a]=t;
}
dijstral(Q);
int temp=inf;
for(i=0;i<N;i++)
if(low[num[i]]<temp)
temp=low[num[i]];
printf("%d\n",temp);
}
return 0;
}
看解题报告的一次迪杰斯特拉算法:
自己多加一个超级源点,把起点集合连接到超级源点上,然后将起点与超级源点的集合的路径长度设为0,这样就称为一个n+1个点的单源最短路算法。。。。。此时,只需要一次Dijkstra就可以求出这个最短长度。
#include<stdio.h>
#include<string.h>
#define inf 0x3f3f3f3f
#define Vex 1001
int map[Vex][Vex];int low[Vex];int M;int visit[Vex];
void dijstral()
{
int i,j,min,pos;
for(i=0;i<=M;i++)
low[i]=map[0][i];
memset(visit,0,sizeof(visit));
low[0]=0;
visit[0]=1;
for(i=0;i<=M;i++)
{
min=inf;pos=0;
for(j=0;j<=M;j++)
{
if(!visit[j]&&low[j]<min){
min=low[j];
pos=j;
}
}
visit[pos]=1;
for(j=0;j<=M;j++){
if(min+map[pos][j]<low[j]){
low[j]=min+map[pos][j];
}
}
}
}
int main()
{
int T,a,b,t,num,N,P,Q,i;
scanf("%d",&T);
for(i=0;i<T;i++){
memset(map,inf,sizeof(map));
scanf("%d%d%d%d",&N,&M,&P,&Q);int i;
for(i=0;i<N;i++){
scanf("%d",&num);
map[0][num]=0;
}
for(num=0;num<P;num++)
{
scanf("%d%d%d",&a,&b,&t);
if(map[a][b]>t)
map[a][b]=map[b][a]=t;
}
dijstral();
printf("%d\n",low[Q]);
}
return 0;
}