poj3259Wormholes

题目链接：http://poj.org/problem?id=3259

用bellman-ford判负环。

#include<cstdio>
#include<cstring>
const int maxn=520;
const int maxe=6000;
const int inf=0x3f3f3f3f;
int t,n,m,w;
int u,v,c;
int dis[maxn];
struct node
{
    int u,v,w,nex;
}e[maxe];

bool bellman()
{
    int flag=1;
    for(int i=0;i<=n;i++){
        dis[i]=inf;
    }
        dis[1]=0;
    for(int i=0;i<n;i++)
    {
        flag=0;
        for(int j=0;j<m;j++)
            if(dis[e[j].v]>dis[e[j].u]+e[j].w){
                dis[e[j].v]=dis[e[j].u]+e[j].w;
            flag=1;
        }

        if(!flag) break;
        if(i==n-1) return 1;

    }
    return 0;

}

int main()
{
    scanf("%d",&t);
    while(t--)
    {

        scanf("%d%d%d",&n,&m,&w);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&c);
            e[i].u=u;
            e[i].v=v;
            e[i].w=c;
            e[i+m].u=v;
            e[i+m].v=u;
            e[i+m].w=c;
        }
        m<<=1;
        for(int i=m;i<w+m;i++)
        {
            scanf("%d%d%d",&u,&v,&c);
            e[i].u=u;
            e[i].v=v;
            e[i].w=-c;
        }
        m+=w;
        if(bellman()) puts("YES");
        else puts("NO");
    }
}