POJ 3259 Wormholes

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

这道题的意思就是 考虑从任意一个点出发 经过任意回路 (普通道路是双向 虫洞是单向并且是负边) 

问回到出发点时 时间是不是小于 出发时的时间

最开始直接想求所有点 到自己点的最短路 

做着做着发现 如果可以的话 那不是没有最短路 因为每次都会更新---->>那不就是负圈

--->>那不就是 判断是否存在负圈

-->>所以直接用Bellman_ford的性质 判断负圈即可

#include <iostream>
#include <stdio.h>
#include <string.h>
#define INF 0x3f3f3f3f
#define MAXV 507
#define MAXE 5800
using namespace std;

struct Edge
{
    int to, cost, next;//向前星 遍历边 然后松弛用向前星同样可以遍历边 只不过按起点而已
}edge[MAXE];

int head[MAXV];
int num = 0;

void Add(int from, int to, int c)
{
    edge[num].cost = c;
    edge[num].to = to;
    edge[num].next = head[from];
    head[from] = num++;
}

int dist[MAXV];
bool bellman_ford(int n)//顶点数
{
    fill(dist, dist+n+3, 0);//全部变为0 查负圈
    for (int k = 0; k < n; k++)//不存在负圈 至多循环n-1次 按这个性质可以判断是否含有负圈
    for (int i = 1; i <= n; i++)
    {
        int t = head[i];
        while (t != -1)
        {
            Edge e = edge[t];
            if (dist[e.to] > dist[i] + e.cost)
            {
                dist[e.to] = dist[i] + e.cost;
                if (k == n-1) return true;//如果第n-1次仍然更新
            }
            t = e.next;
        }
    }
    return false;
}
int main()
{
    int F;
    freopen("in.txt", "r", stdin);
    scanf("%d",&F);
    while(F--)
    {
        int N, M, W;
        num = 0;
        memset(edge, -1, sizeof(edge));
        memset(head, -1, sizeof(head));
        scanf("%d%d%d", &N, &M, &W);
        for (int i = 0;i < M; i++)
        {
            int from, to, cost;
            scanf("%d%d%d", &from, &to, &cost);
            Add(from, to, cost);
            Add(to, from, cost);//双向边
        }
        for (int i = 0; i < W; i++)
        {
            int from, to, cost;
            scanf("%d%d%d", &from, &to, &cost);
            Add(from, to, -cost);
        }
        if (bellman_ford(N)) cout << "YES" << endl;
        else cout << "NO" << endl;
    }
}