SPFA(最短路问题)

SPFA算法介绍

SPFA算法是求解单源最短路径问题的一种算法，由理查德·贝尔曼（Richard Bellman） 和 莱斯特·福特 创立的。有时候这种算法也被称为 Moore-Bellman-Ford 算法，因为 Edward F. Moore 也为这个算法的发展做出了贡献。它的原理是对图进行V-1次松弛操作，得到所有可能的最短路径。其优于迪杰斯特拉算法的方面是边的权值可以为负数、实现简单，缺点是时间复杂度过高，高达 O(VE) 其中V是节点数量，E是边数量。但算法可以进行若干种优化，提高了效率。

SPFA算法思路

我们用数组dis记录每个结点的最短路径估计值，用二维数组来存储图E。我们采取的方法是动态逼近法：设立一个队列用来保存待优化的结点，优化时每次取出队首结点u，并且用u点当前的最短路径估计值对离开u点所指向的结点v进行松弛操作，如果v点的最短路径估计值有所调整，且v点不在当前的队列中，就将v点放入队尾。这样不断从队列中取出结点来进行松弛操作，直至队列空为止
注意：如果题目明确说明存在负环，那么就必须用SPFA算法来解决，这时候我们必须另开一个数组用于判断i号点到j号点的最短路径的步数是否<n（n为节点数），否则每次经过负环，他的权值都会减少，这样会导致无限循环，无法算出结果

代码及讲解

#include <bits/stdc++.h>
#define lowbit(x) ((x)&(-x))
#define int long long
#define endl '\n'
#define PII pair<int,int> 
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
int m,n,d[5010],st[5010],cnt[5010];
queue<int>q;
vector<PII>vt[5010];
bool spfa(int s)
{
	memset(d,0x3f,sizeof d);
	memset(st,0,sizeof st);
	memset(cnt,0,sizeof cnt);
	d[s]=0;st[s]=1;q.push(s);
	while(q.size())
	{
		int t=q.front();
		st[t]=0;q.pop();
		for(auto i:vt[t])
		{
			int a=i.first,b=i.second;
			if(d[a]>d[t]+b)
			{
				d[a]=d[t]+b;
				cnt[a]=cnt[t]+1;
				if(cnt[a]>=n)return 1;
				if(st[a]==0)st[a]=1,q.push(a);
			}
		}
	}
	return 0;
}
signed main()
{
	IOS
	int T=1;
	cin>>T;
	while(T--)
	{
		for(int i=0;i<=5010;i++)
		vt[i].clear();
	    cin>>n>>m;
		for(int i=1;i<=m;i++)
		{
			int u,v,w;
			cin>>u>>v>>w;
			vt[u].push_back({v,w});
		    if(w>=0) vt[v].push_back({u,w});
		}
		int t=spfa(1);
		if(t==1)
		cout<<"YES"<<endl;
		else
		cout<<"NO"<<endl;
	}
	return 0;
}

模板链接