POJ 3259 Wormholes

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

 

题意：John的农场里field块地，path条路连接两块地，hole个虫洞，虫洞是一条单向路，不但会把你传送到目的地，而且时间会倒退Ts。我们的任务是知道会不会在从某块地出发后又回来，看到了离开之前的自己。

 

思路：用bellman-ford 判断有没有负权回路，如果有他就能看到自己。 不过，我认为应该判断每个点有没有负权回路，而不仅仅只判断第一个点就行了。

AC代码

#include <iostream>

#include <cstdio>

#include <cstring>

using namespace std;

const int VM=520;

const int EM=2600;

const int INF=0x3f3f3f3f;

struct Edge{

    int u,v;

    int cap;

}edge[EM<<1];

int n,m,k;

int cnt,dis[VM];

int 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<=cnt-1;j++)

        {

            if(dis[edge[j].v]>dis[edge[j].u]+edge[j].cap)

            {

                dis[edge[j].v]=dis[edge[j].u]+edge[j].cap;

            }

        }

    }

    for(j=1;j<cnt-1;j++)

    {

        if(dis[edge[j].v]>dis[edge[j].u]+edge[j].cap)

        {

            return 0;

        }

    }

    return 1;

}

int main()

{

    int t;

    scanf("%d",&t);

    while(t--)

    {

        scanf("%d%d%d",&n,&m,&k);

        cnt=1;

        int u,v,w;

        while(m--)

        {

            scanf("%d%d%d",&u,&v,&w);

            edge[cnt].u=u;

            edge[cnt].v=v;

            edge[cnt++].cap=w;

            edge[cnt].u=v;

            edge[cnt].v=u;

            edge[cnt++].cap=w;

        }

        while(k--)

        {

            scanf("%d%d%d",&u,&v,&w);

            edge[cnt].u=u;

            edge[cnt].v=v;

            edge[cnt++].cap=-w;

        }

        int ans=Bellman();

        if(ans==0){

            printf("YES\n");

        }

        else

        {

            printf("NO\n");

        }

    }

    return 0;

}