从Bellman_Ford到spfa（Bellman_Ford的队列优化）

最新推荐文章于 2024-11-02 15:25:13 发布

河林山

最新推荐文章于 2024-11-02 15:25:13 发布

阅读量225

点赞数 1

分类专栏：最短和最小生成树算法文章标签：算法

本文链接：https://blog.csdn.net/Helinshan/article/details/123941797

版权

最短和最小生成树算法专栏收录该内容

15 篇文章 1 订阅

订阅专栏

题目链接

代码1(超时)：Bellman_Ford原始算法

#include<bits/stdc++.h>
using namespace std;
const int N=20020;
const int M=200020;
const int INF=0x3f3f3f3f;
int u[M],v[M],w[M];//边的信息 
int dis[N];//源点到别的点最短路
int n,m;
void Bellman_Ford()
{ 
	memset(dis,INF,sizeof(dis));
	dis[1]=0;//源点 
	for(int k=1;k<=n-1;k++)
		for(int i=1;i<=m;i++)
			if(dis[v[i]]>dis[u[i]]+w[i])
				dis[v[i]]=dis[u[i]]+w[i];
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;i++)
		scanf("%d %d %d",&u[i],&v[i],&w[i]);
	Bellman_Ford();
	for(int i=2;i<=n;i++)
		printf("%d\n",dis[i]);	
	return 0;
}

代码2(AC)：Bellman_Ford的一点优化

#include<bits/stdc++.h>
using namespace std;
const int N=20020;
const int M=200020;
const int INF=0x3f3f3f3f;
int u[M],v[M],w[M];//边的信息 
int dis[N];//源点到别的点最短路
int n,m;
void Bellman_Ford()
{ 
	int i,k,flag; 
	memset(dis,INF,sizeof(dis));
	dis[1]=0;//源点 
	for(k=1;k<=n-1+10;k++)
	{
		flag=0;
		for(i=1;i<=m;i++)
		{
			if(dis[v[i]]>dis[u[i]]+w[i])
			{
				dis[v[i]]=dis[u[i]]+w[i];
				flag=1;
			}
		}
		if(!flag)	break;
	}
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;i++)
		scanf("%d %d %d",&u[i],&v[i],&w[i]);
	Bellman_Ford();
	for(int i=2;i<=n;i++)
		printf("%d\n",dis[i]);	
	return 0;
}

代码3(AC)：Bellman_Ford的队列优化（spfa）

#include<bits/stdc++.h>
using namespace std;
const int N=20020;
const int M=200020;
const int INF=0x3f3f3f3f;
int first[N],nx[M],to[M],s[M],head;
int dis[N],book[N];//源点到别的点最短路
int n,m;
void init()
{
	head=1;
	memset(first,-1,sizeof(first));
}
void add(int x,int y,int z)
{
	nx[head]=first[x];
	to[head]=y;
	s[head]=z;
	first[x]=head++;
}
void spfa(int x)
{
	queue<int> Q;
	memset(book,0,sizeof(book));
	memset(dis,INF,sizeof(dis));
	dis[x]=0;
	Q.push(x);
	book[x]=1;
	int u,v,i;
	while(!Q.empty())
	{
		u=Q.front();
		Q.pop();
		book[u]=0;
		for(i=first[u];i!=-1;i=nx[i])
		{
			v=to[i];
			if(dis[v]>dis[u]+s[i])
			{
				dis[v]=dis[u]+s[i];
				if(!book[v])
				{
					book[v]=1;
					Q.push(v);
				}
			}
		}
	}
}
int main()
{
	init();
	int u,v,w,i;
	scanf("%d %d",&n,&m);
	for(i=1;i<=m;i++)
	{
		scanf("%d %d %d",&u,&v,&w);
		add(u,v,w);
	}
	spfa(1);
	for(i=2;i<=n;i++)
		printf("%d\n",dis[i]);
	return 0;
}