POJ 3259 Wormholes（Bellman-Ford）

http://poj.org/problem?id=3259

题意：
有一些普通的洞和虫洞，每个洞都有经过的时间，虫洞的时间是负的，也就是时光倒流，问是否能回到出发时的时间。

思路：

贝尔曼-福特算法判断负环。

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<set>
#include<map>
using namespace std;

const int maxn = 5000 + 5;
const int INF = 10000 + 5;

int n, num1, num2;
int cnt;
int d[maxn];

struct node
{
    int s;
    int e;
    int w;
}g[maxn];

bool bellman()
{
    memset(d, INF, sizeof(d));
    for (int i = 0; i < n; i++)
    {
        bool flag = false;
        for (int j = 0; j < cnt; j++)
        {
            if (d[g[j].e]>d[g[j].s] + g[j].w)
            {
                d[g[j].e] = d[g[j].s] + g[j].w;
                flag = true;
            }
        }
        if (!flag)  break;
    }
    for (int j = 0; j < cnt;j++)
    if (d[g[j].e]>d[g[j].s] + g[j].w)
        return true;
    return false;
}

int main()
{
    //freopen("D:\txt.txt", "r", stdin);
    int T;
    int a, b, t;
    scanf("%d", &T);
    while (T--)
    {
        cnt = 0;
        scanf("%d%d%d", &n, &num1, &num2);
        for (int i = 0; i < num1; i++)
        {
            scanf("%d%d%d", &a, &b, &t);
            g[cnt].s = g[cnt + 1].e = a;
            g[cnt].e = g[cnt + 1].s = b;
            g[cnt].w = g[cnt + 1].w = t;
            cnt += 2;
        }
        for (int i = 0; i < num2; i++)
        {
            scanf("%d%d%d", &a, &b, &t);
            g[cnt].s = a;
            g[cnt].e = b;
            g[cnt].w = -t;
            cnt++;
        }
        if (bellman())
            printf("YES
");
        else
            printf("NO
");
    }
    return 0;
}