洛谷 P3385 判负环dfs

原题戳

emmmmmm，200000万条边居然能T掉，果断学了新的算法，dfs-spfa(),判负环；

因为只需要判负环，所以初始dis数组为0，这时如果没有负环的话，要么不会更新，要么只更新一遍，效率比spfa快好多

前向星写法

By Acer.Mo

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int INF=1<<29,M=500500;
int dis[M],vis[M],cnt[M];
int n,m,tot,flag;
int head[M],next[M],go[M],cost[M];
void add(int a,int b,int c)
{
    go[tot]=b;
    cost[tot]=c;
    next[tot]=head[a];
    head[a]=tot++;
    
    if (c>=0)
    {
        go[tot]=a;
        cost[tot]=c;
        next[tot]=head[b];
        head[b]=tot++;
    }
    return ;
}
void dfspfa(int x)
{
    if (flag) return ;
    vis[x]=1;
    for (int i=head[x];i!=-1;i=next[i])
    {
        int to=go[i];
        int pay=cost[i];
        if (dis[to]>dis[x]+pay)
        {
            dis[to]=dis[x]+pay;
            if (vis[to]) 
            {
                flag=1;
                return ;
            }
            else dfspfa(to);
        }
    }
    vis[x]=0;
    return ;
}
int main()
{
    int t;
    scanf("%d",&t);
    while (t--)
    {
        scanf("%d %d",&n,&m);
        fill(head,head+n+1,-1);
        fill(go,go+n+1,0);
        fill(cost,cost+n+1,0);
        fill(next,next+n+1,0);
        fill(dis,dis+n+1,0);
        fill(vis,vis+n+1,0);
        tot=0;flag=0;
        for (int i=1;i<=m;i++)
        {
            int a,b,c;
            scanf("%d %d %d",&a,&b,&c);
            add(a,b,c);
        }
        for (int i=1;i<=n;i++) 
        {
            dfspfa(i);
            if (flag) break;
        }
        if (flag) printf("YE5\n");
        else printf("N0\n");
    }
    return 0;
}

STL写法

By Acer.Mo

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const int INF=1<<29,M=500500;
int dis[M],vis[M],cnt[M];
int n,m,flag;
struct edge
{
    int cost;
    int to;
};
vectorv[M];
void dfspfa(int x)
{
    if (flag) return ;
    vis[x]=1;
    for (int i=0;idis[x]+pay)
        {
            dis[to]=dis[x]+pay;
            if (vis[to]) 
            {
                flag=1;
                return ;
            }
            else dfspfa(to);
        }
    }
    vis[x]=0;
    return ;
}
int main()
{
    int t;
    scanf("%d",&t);
    while (t--)
    {
        scanf("%d %d",&n,&m);flag=0;
        fill(dis,dis+n+1,0);
        fill(vis,vis+n+1,0);
        for (int i=1;i<=m;i++)
        {
            int a,b,c;edge now;
            scanf("%d %d %d",&a,&b,&c);
            now.cost=c;
            now.to=b;
            v[a].push_back(now);
            if (c>=0)
            {
                now.to=a;
                v[b].push_back(now);
            }
        }
        for (int i=1;i<=n;i++)
        {
        	dfspfa(i);
        	if (flag) break;
        }
        if (flag) printf("YE5\n");
        else printf("N0\n");
        for (int i=1;i<=n;i++)
        {
            v[i].clear();
        }
    }
    return 0;
}