POJ 3259-Wormholes

http://poj.org/problem?id=3259
　　题意大致是N个点， M条正常的边（双向）边权为正，W个虫洞，边权为负，想要

我们判断回路中是否存在一个负圈。学习了bellman-ford算法来求负环，松弛结束后判断是否

存在最短路，不存在则有负环。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

const int maxn = 1 << 9, inf = 0x3f3f3f3f;
int dist[maxn];
int N, M, W, es;

struct Edge
{
    int u, v, w;
}e[maxn * maxn];

void addedge(int u, int v, int w)
{
    e[es].u = u, e[es].v = v, e[es ++].w = w;
}

void readgraph()
{
    int u, v, w;
    es = 0;
    scanf("%d%d%d", &N, &M, &W);
    for(int i = 0; i < M; i ++)
    {
        scanf("%d%d%d", &u, &v, &w);
        addedge(u, v, w);
        addedge(v, u, w);
    }
    for(int i = 0; i < W; i ++)
    {
        scanf("%d%d%d", &u, &v, &w);
        addedge(u, v, -w);
    }
}

void relax(int es)
{
    int u = e[es].u, v = e[es].v, w = e[es].w;
    if(dist[v] > dist[u] + w)
        dist[v] = dist[u] + w;
}

bool bellman(int src)
{
    for(int i = 1; i < N; i ++)
        dist[i] = inf;
    dist[src] = 0;
    for(int i = 0; i < N - 1; i ++)
    {
        for(int j = 0; j < es; j ++)
        {
            relax(j);
        }
    }
    for(int i = 0; i < es; i ++)
    {
        if(dist[e[i].v] > dist[e[i].u] + e[i].w)
            return false;
    }
    return true;
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T --)
    {
        readgraph();
        if(!bellman(0))
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}

 

 

下面就是这个算法的过程： 

Bellman-Ford(G,w,s) ：boolean //图G ，边集 函数 w ，s为源点

 

　　1 for each vertex v ∈ V（G） do //初始化 1阶段

 

　　2 d[v] ←+∞

 

　　3 d[s] ←0; //1阶段结束

 

　　4 for i=1 to |v|-1 do //2阶段开始，双重循环。

 

　　5 for each edge（u,v） ∈E(G) do //边集数组要用到，穷举每条边。

 

　　6 If d[v]> d[u]+ w(u,v) then //松弛判断

 

　　7 d[v]=d[u]+w(u,v) //松弛操作 2阶段结束

 

　　8 for each edge（u,v） ∈E(G) do

 

　　9 If d[v]> d[u]+ w(u,v) then

 

　　10 Exit false

 

　　11 Exit true