[模板]负环

在博大精深的图论算法中有这样一个神奇的存在!

当我们求最短路时,往往会发现有边权为负的情况存在,这时候我们的dijksra便不能很好的胜任他的职务了。

那么伟大的spfa算法就出现了(至于同学们在刚接触这个算法的时候,一定会听到:关于SPFA,他死了,这是因为dfs优化下的spfa时间复杂度极优,亲测是dijksta的10分之1左右,但是它有着明显的缺陷,

容易被出题人的极端数据卡掉,所以我们今天讲的是常用的bfs的spfa)

我们用一个队列储存所有的点,用vis数组储存他是访问过了,剩下的就是松弛操作。

我一般爱把其写为bool类型,如果节点的访问次数>n即有负环,返回false

代码如下:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<queue>
#define maxn 120000
using namespace std;

inline int read()
{
    char c=getchar();int re=0,f=1;
    while('0'>c||c>'9')
    {
        if(c=='-')
        {
            f=-1;
        }
        c=getchar();
    }
    while('0'<=c&&c<='9')
    {
        re=re*10+c-'0';
        c=getchar();
    }
    return re*f;
}

int head[maxn],n,m,cnt;
int vis[maxn],dis[maxn],num[maxn];

struct node
{
    int next,to,w;
}e[maxn];

inline void add(int x,int y,int z)
{
    e[++cnt].to=y;
    e[cnt].w=z;
    e[cnt].next=head[x];
    head[x]=cnt;
}


bool spfa(int s)
{
    memset(vis,0,sizeof(vis));
    memset(dis,0x3f3f3f3f,sizeof(dis));
    memset(num,0,sizeof(num));
    queue<int> q;
    q.push(s);
    vis[s]=1;
    dis[s]=0;
    
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        //if(num[u]>=n)return 1; 
        for(int i=head[u];i;i=e[i].next)
        {
            int v=e[i].to;
            int w=e[i].w;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                if(!vis[v])
                {
                    q.push(v);
                    vis[v]=1;
                    ++num[v];
                    if(num[v]>n)
                    {
                        return 1;
                    }
                }
                
            }
        }
    }
    return 0;
}

int main()
{
    int t=read();
    while(t--)
    {
        n=read(),m=read();
        memset(head,0,sizeof(head));
        cnt=0;
        for(int i=1;i<=m;i++)
        {
            int x,y,z;
            x=read(),y=read(),z=read();
            if(z>=0)
            {
                add(x,y,z);
                add(y,x,z);
            }
            else
            {
                add(x,y,z);
            }
            
        }
        if(spfa(1))
            {
            cout<<"YE5"<<endl;
        }
        else{
            cout<<"N0"<<endl;
        }
    }
    return 0;
}

但是在题目中没有强调负边权的情况下,我们还是要用堆优化的dijksta啊!

——2021届董驰原

猜你喜欢

转载自www.cnblogs.com/btjzoi/p/11824178.html
今日推荐