洛谷 P3385 【模板】负环 spfa 负环

题目链接：

​​https://www.luogu.org/problemnew/show/P3385​​

思路：

1：spfa判断负环

2：当某个点松弛超过n次后，便可以说明有负环

#include <bits/stdc++.h>

using namespace std;
const int maxn=2001;
vector<pair<int,int> >e[maxn];
int n,m,f,d[maxn],ing[maxn],a,b,c,cou[maxn];

void init()
{
    for(int i=1;i<=n;i++)
    {
        ing[i]=0;
        d[i]=1e9;
        e[i].clear();
        cou[i]=0;
    }
}

string spfa(int s)
{
    queue<int>q;
    q.push(s);
    cou[s]++;
    d[s]=0;
    ing[s]=1;
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        ing[now]=0;
        for(int i=0;i<e[now].size();i++)
        {
            int v=e[now][i].first;
            if(d[v]>d[now]+e[now][i].second)
            {
                d[v]=d[now]+e[now][i].second;
                if(ing[v])
                    continue;
                q.push(v);
                cou[v]++;
                if(cou[v]>n)
                    return "YE5";
                ing[v]=1;
            }
        }
    }
    return "N0";
}

int main()
{
    cin>>f;
    while(f--)
    {
        cin>>n>>m;
        init();
        for(int i=0;i<m;i++)
        {
             cin>>a>>b>>c;
             if(c>=0)
             {
                  e[a].push_back(make_pair(b,c));
                  e[b].push_back(make_pair(a,c));
             }
             else
             {
                 e[a].push_back(make_pair(b,c));
             }
        }
        cout<<spfa(1)<<endl;
    }
    return 0;
}