洛谷P3385判负环——spfa

题目：https://www.luogu.org/problemnew/show/P3385

两种方法，dfs和bfs；

一开始写的dfs，要把dis数组初值赋成0，这样从一个连着负边的点开始搜；

在一个负环上，一定会有一个点，从它开始绕环走，dis值一直为负，根据这个找环；

但是数据太强了，过不了：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int const MAXN=,MAXM=;
int T,n,m,head[MAXN],ct,dis[MAXN];
bool vis[MAXN],f;
struct N{
    int to,next,w;
    N(int t=,int n=,int w=):to(t),next(n),w(w) {}
}edge[MAXM];
void add(int x,int y,int z)
{
    edge[++ct]=N(y,head[x],z);head[x]=ct;
}
int rd()
{
    int x=,f=;
    char ch=getchar();
    while(ch<''||ch>'')
    {
        if(ch=='-')f=-;
        ch=getchar();
    }
    while(ch>=''&&ch<='')x=x*+ch-'',ch=getchar();
    return x*f;
}
void dfs(int x)
{
    if(f)return;
    vis[x]=;
    for(int i=head[x],u;i;i=edge[i].next)
    {
        u=edge[i].to;
        if(f)return;
        if(dis[u]>dis[x]+edge[i].w)
        {
//            printf("x=%d u=%d vis[u]=%d\n",x,u,vis[u]);
            if(vis[u])
            {
                f=;return;
            }
            dis[u]=dis[x]+edge[i].w;
            dfs(u);
            if(f)return;
        }
    }
    vis[x]=;//!
}
int main()
{
    T=rd();
    while(T--)
    {
        n=rd();m=rd();
        ct=;f=;
        memset(head,,sizeof head);
        for(int i=,x,y,z;i<=m;i++)
        {
            x=rd();y=rd();z=rd();
            add(x,y,z);
            if(z>=)add(y,x,z);
        }
        memset(vis,,sizeof vis);
        memset(dis,,sizeof dis);
        for(int i=;i<=n;i++)
        {
            dfs(i);
            if(f)break;
        }
        if(f)printf("YE5\n");
        else printf("N0\n");
    }
    return ;
}

dfs

于是用bfs，根据最短路边数来判断，若边数>=n则有负环；

bfs的话还有一种判断方式，是根据被松弛次数，若>=n则有负环，但不如上面那个优；

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
//queue<int>q;
int const MAXN=,MAXM=,inf=;
int T,n,m,head[MAXN],ct,dis[MAXN],cnt[MAXN],que[MAXN],h,t;
bool vis[MAXN];
struct N{
    int to,next,w;
    N(int t=,int n=,int w=):to(t),next(n),w(w) {}
}edge[MAXM];
inline void add(int x,int y,int z){edge[++ct]=N(y,head[x],z);head[x]=ct;}
inline int rd()
{
    int x=,f=;
    char ch=getchar();
    while(ch<''||ch>'')
    {
        if(ch=='-')f=-;
        ch=getchar();
    }
    while(ch>=''&&ch<='')x=x*+ch-'',ch=getchar();
    return x*f;
}
inline bool spfa()
{
//    while(q.size())q.pop();
    memset(que,,sizeof que);h=;t=;
    memset(dis,0x3f,sizeof dis);
    memset(cnt,,sizeof cnt);
    memset(vis,,sizeof vis);
//    q.push(1);
    vis[]=;dis[]=;que[t]=;
//    while(q.size())
    while(h!=t+)
    {
//        int x=q.top();vis[x]=0;q.pop();
        int x=que[h++];vis[x]=;
        if(h==inf)h=;
        for(int i=head[x],u;i;i=edge[i].next)
            if(dis[u=edge[i].to]>dis[x]+edge[i].w)
            {
                cnt[u]=cnt[x]+;
                if(cnt[u]>=n)return ;
                dis[u]=dis[x]+edge[i].w;
//                if(!vis[u])vis[u]=1,q.push(u);
                if(!vis[u])
                {
                    vis[u]=;
                    t++;
                    if(t==inf)t=;
                    que[t]=u;
                }
            }
    }
    return ;
}
int main()
{
    T=rd();
    while(T--)
    {
        n=rd();m=rd();
        ct=;
        memset(head,,sizeof head);
        for(int i=,x,y,z;i<=m;i++)
        {
            x=rd();y=rd();z=rd();
            add(x,y,z);
            if(z>=)add(y,x,z);
        }
        if(spfa())printf("YE5\n");
        else printf("N0\n");
    }
    return ;
}