poj3259 Wormholes（bellman负权处理）

思路：用bellman建立无向边，然后将虫洞的边置为负值求最短路，在进行n-1次松弛后，判断是否还能继续松弛。如果可以，则有负权回路，则可以在出发前回到原地，输出YES，否则，NO；

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#define N 6000//建立无向边时注意N的范围相应的扩大为二倍
int u[N],v[N],w[N],dis[N];
int n,m,W;
int inf=9999999;

void bellman()
{
    int i,j;
    for(i=1; i<=n; i++)
        dis[i]=inf;
    dis[1]=0;
    for(i=1; i<n; i++)
        for(j=1; j<=2*m+W; j++)
        {
            if(dis[u[j]]<inf&&dis[v[j]]>dis[u[j]]+w[j])
                dis[v[j]]=dis[u[j]]+w[j];
        }
}
int main()
{
    int t,c,a,b,i;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&W);
        for(i=1; i<=2*m; i+=2)//建立无向边，正着反着各建一条边
        {
            scanf("%d%d%d",&a,&b,&w[i]);
            u[i]=a;v[i]=b;
            v[i+1]=a,u[i+1]=b;w[i+1]=w[i];
        }

        for(i=2*m+1; i<=W+2*m; i++)
        {
            scanf("%d%d%d",&u[i],&v[i],&c);
            w[i]=-c;//由虫洞回去的边置为负值
        }
        bellman();
        int flag=0;
        for(i=1; i<=2*m+W; i++)
        {
            if(dis[u[i]]<inf&&dis[v[i]]>dis[u[i]]+w[i])
                flag=1;
        }
        if(flag==1)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}

建立无向边时候一定要注意N的范围相应的扩大!!