用SPFA判断是否存在负环

题目描述

暴力枚举/SPFA/Bellman-ford/奇怪的贪心/超神搜索

输入输出格式

输入格式：

 

第一行一个正整数T表示数据组数，对于每组数据：

第一行两个正整数N M，表示图有N个顶点，M条边

接下来M行，每行三个整数a b w，表示a->b有一条权值为w的边（若w<0则为单向，否则双向）

 

输出格式：

 

共T行。对于每组数据，存在负环则输出一行"YE5"(不含引号)，否则输出一行"N0"(不含引号)。

 

负环指的是权值和为负数的环，用SPFA加上DFS做比较方便，如果用BFS来做就要遍历的话判断条件是入队n次，但不好操作。用DFS就好写多了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stack>
#include<stdlib.h>
#include<map>
#include<vector>
#include<queue>
#include<set>
#include<ctype.h>
#define zuida 100000
using namespace std;
#define maxn 200005
struct node
{
    int to;
    int next;
    int val;
}bb[maxn];
int head[maxn];
int tot;
int vis[maxn];
int dis[maxn];
int flag;
void add(int a,int b,int c)
{
    bb[++tot].next=head[a];
    bb[tot].to=b;
    bb[tot].val=c;
    head[a]=tot;
}
void spfa(int u)
{
    if(flag)return;
    vis[u]=1;
    for(int i=head[u];i;i=bb[i].next)
    {
        if(flag)return;
        int v=bb[i].to;
        if(dis[v]>bb[i].val+dis[u])
        {

            if(vis[v])
            {
                flag=1;
                cout<<v<<endl;
                return ;
            }
            dis[v]=dis[u]+bb[i].val;
            spfa(v);
        }
    }
    vis[u]=0;
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        tot=0;
        flag=0;
        int n,m;
        cin>>n>>m;
        memset(dis,0,sizeof(dis));
        memset(vis,0,sizeof(vis));
        memset(head,0,sizeof(head));
        dis[1]=0;
        int aa,bb,cc;
        for(int i=1;i<=m;i++)
        {
            cin>>aa>>bb>>cc;
            add(aa,bb,cc);
            if(cc>=0)add(bb,aa,cc);
        }
        for(int i=1;i<=n;i++)
        {
            spfa(i);
            if(flag)break;

        }
        if(flag)cout<<"YE5"<<endl;
        else cout<<"N0"<<endl;
    }
    return 0;
}