本蒟蒻在做负环的模板题时调了好久也只有36分,请各位大佬帮忙看一看我的代码哪里有问题,谢谢!
#include <bits/stdc++.h>
#define inf 0x3f3f3f
#define N 5000
using namespace std;
int n,m,t;
int dis[N],cnt[N];
bool vis[N];
struct node{int to,dis;};
queue<int>q;
vector<node>g[N];
void work()
{
cin>>n>>m;
int u,v,w;
for(int i=1;i<=m;i++)
{
cin>>u>>v>>w;
g[u].push_back((node){v,w});
if(w>=0)g[v].push_back((node){u,w});
}
memset(dis,inf,sizeof(dis));
memset(vis,0,sizeof(vis));
memset(cnt,0,sizeof(cnt));
while(!q.empty())q.pop();
dis[1]=0;vis[1]=1;q.push(1);
while(!q.empty())
{
int u=q.front();
vis[u]=0;
q.pop();
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i].to;
if(dis[v]>dis[u]+g[u][i].dis)
{
dis[v]=dis[u]+g[u][i].dis;
cnt[v]=cnt[u]+1;
if(cnt[v]>=n)
{
cout<<"YES"<<endl;
return;
}
if(!vis[v])
{
vis[v]=1;q.push(v);
}
}
}
}
cout<<"NO"<<endl;
}
int main()
{
cin>>t;
while(t--)work();
return 0;
}