洛谷 P3385 判负环dfs

原题戳

emmmmmm，200000万条边居然能T掉，果断学了新的算法，dfs-spfa(),判负环；
因为只需要判负环，所以初始dis数组为0，这时如果没有负环的话，要么不会更新，要么只更新一遍，效率比spfa快好多
前向星写法By Acer.Mo

#include<cmath>
#include<queue>
#include<stack>
#include<deque>
#include<vector>
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
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<cmath>
#include<queue>
#include<stack>
#include<deque>
#include<vector>
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
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;
};
vector<edge>v[M];
void dfspfa(int x)
{
if (flag) return ;
vis[x]=1;
for (int i=0;i<v[x].size();i++)
{
int to=v[x][i].to;
int pay=v[x][i].cost;
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);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;
}