$\huge \color{orange}{成仙之路->}$ $\huge \color{purple}{算法基础课题解}$
思路:
1. 因为如果存在负环,dist就没必要初始化
2. cnt[i] 表示该点到某点的边数,只要 cnt[i] == n 就说明存在负环
3. 把所有点都放入队列,如果存在负环就一定能遍历到
完整代码
#include<bits/stdc++.h>
using namespace std;
const int N = 2010, M = 10010;
int n,m;
int h[N],e[M],w[M],ne[M],idx;
int dist[N],cnt[N];
bool st[N];
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
bool spfa()
{
//把所有点都放入队列,这样就一定能遍历到负环
queue<int> q;
for(int i=1;i<=n;i++)
{
q.push(i);
st[i]=true;
}
while(q.size())
{
auto t=q.front();
q.pop();
st[t]=false;
for(int i=h[t];~i;i=ne[i])
{
int j=e[i];
if(dist[t]+w[i]<dist[j])
{
//更新距离,并更新该点到某点的边数
dist[j]=dist[t]+w[i];
cnt[j]=cnt[t]+1;
//当边数达到n,则说明存在负环
if(cnt[j]==n) return true;
if(!st[j])
{
q.push(j);
st[j]=true;
}
}
}
}
return false;
}
int main()
{
cin>>n>>m;
memset(h,-1,sizeof h);
while(m--)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
}
cout<<(spfa() ? "Yes" : "No")<<endl;
return 0;
}