【BZOJ 1715】 [Usaco2006 Dec]Wormholes 虫洞

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1715
SPFA判负权环
我的模板库

#include<iostream>#include<algorithm>#include<cstring>#include<cmath>#include<cstdio>#include<queue>const int maxe(3000);const int maxv(800);using namespace std;int ecnt,n,dis[maxv],sum[maxv];bool vis[maxv];int head[maxv];struct edge{    int next,to,dis;}ed[maxe];queue<int> q;void add_edge(int from,int to,int dis){    ed[++ecnt].next=head[from];    ed[ecnt].to=to;    ed[ecnt].dis=dis;    head[from]=ecnt;}void init(){    int m,w;    scanf("%d%d%d",&n,&m,&w);    int a1,a2,a3;    for(int i=1;i<=m;i++)    {        scanf("%d%d%d",&a1,&a2,&a3);        add_edge(a1,a2,a3);        add_edge(a2,a1,a3);    }    for(int i=1;i<=w;i++)    {        scanf("%d%d%d",&a1,&a2,&a3);        add_edge(a1,a2,-a3);//      add_edge(a2,a1,-a3);    }    return;}bool SPFA(){    memset(vis,0,sizeof(vis));    memset(dis,0x3f,sizeof(dis));    memset(sum,0,sizeof(sum));    dis[1]=0;    q.push(1);    vis[1]=1;       while(!q.empty())    {        int t=q.front();        vis[t]=0;        q.pop();        for(int i=head[t];i;i=ed[i].next)        {            int et=ed[i].to;            int d=ed[i].dis;            if(d+dis[t]<dis[et])            {                if(++sum[et]>n) return false;                dis[et]=d+dis[t];                if(!vis[et])                {                    q.push(et);                    vis[et]=1;                }            }        }    }    return 1;}int main(){       int tt;    scanf("%d",&tt);    for(int i=1;i<=tt;i++)    {        init();        if(SPFA()) printf("NO\n");        else printf("YES\n");       }     return 0;}