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;
}