POJ3259 Wormholes(最短路)

题目链接。

分析：

本题的关键的关键是——看懂题。。。

本质就是求该图是否存在负环。也就是如何求出一个图是否含有负环。

要注意的是输入该题中的path是双向的，wormhole是单向的。

Bellman_Ford:

#include <stdio.h>

#define MAXN 520
#define MAXM 6000

const int INF = (1<<26);

struct node{
    int u, v, w;
}edge[MAXM];

int dis[MAXN], top;

int bellman_ford(int n){
    int i, j, u, v, w;

    for(i=1; i<=n; i++) dis[i] = INF;
    dis[1] = 0;
    for(i=0; i<n-1; i++){
        for(j=0; j<top; j++){
            u = edge[j].u;
            v = edge[j].v;
            w = edge[j].w;
            if(dis[u] < INF && dis[u]+w < dis[v]){
                dis[v] = dis[u]+w;
            }
        }
    }

    for(j=0; j<top; j++){
        u = edge[j].u;
        v = edge[j].v;
        w = edge[j].w;
        if(dis[u] < INF && dis[u]+w < dis[v]){
            return 0;
        }
    }
    return 1;
}

void add(int u, int v, int w){
    edge[top].u = u;
    edge[top].v = v;
    edge[top++].w = w;
}

int main(){
    int T, n, m, l, i, u, v, w;

    scanf("%d", &T);
    while(T--){
        top = 0;
        scanf("%d %d %d", &n, &m, &l);
        for(i=0; i<m; i++){
            scanf("%d %d %d", &u, &v, &w);
            add(u, v, w);
            add(v, u, w);
        }
        for(i=0; i<l; i++){
            scanf("%d %d %d", &u, &v, &w);
            add(u, v, -w);
        }
        if(!bellman_ford(n)) printf("YES\n");
        else printf("NO\n");
    }

    return 0;
}

SPFA：

#include <cstdio>
#include <queue>

using namespace std;

#define MAXN 520
#define MAXM 6000

const int INF = (1<<26);

struct node{
    int v, w;
    int next;
}edge[MAXM];

queue<int> q;
int dis[MAXN], head[MAXN], top, inq[MAXN], used[MAXN];


void add(int u, int v, int w){
    edge[top].v = v;
    edge[top].w = w;
    edge[top].next = head[u];
    head[u] = top++;
}

int spfa(int n){
    for(int i=1; i<=n; i++) dis[i] = INF;
    dis[1] = 0;
    q.push(1);
    used[1]++;
    while(!q.empty()){
        int x = q.front(); q.pop();
        inq[x] = 0;
        for(int u = head[x]; u != -1; u = edge[u].next){
            if(dis[edge[u].v] > dis[x]+edge[u].w){
                dis[edge[u].v] = dis[x]+edge[u].w;
                if(!inq[edge[u].v]){
                    inq[edge[u].v] = 1;
                    if(++used[edge[u].v] > n-1) return 0;
                    q.push(edge[u].v);
                }
            }
        }
    }
    return 1;
}

void Init(int n){
    top = 0;
    for(int i=1; i<=n; i++){
        inq[i] = 0;
        used[i] = 0;
        head[i] = -1;
    }
}

int main(){
    int T, n, m, l, i, u, v, w;

    scanf("%d", &T);

    while(T--){
        scanf("%d %d %d", &n, &m, &l);

        Init(n);

        for(i=0; i<m; i++){
            scanf("%d %d %d", &u, &v, &w);
            add(u, v, w);
            add(v, u, w);
        }
        for(i=0; i<l; i++){
            scanf("%d %d %d", &u, &v, &w);
            add(u, v, -w);
        }

        if(!spfa(n)) printf("YES\n");
        else printf("NO\n");
    }

    return 0;
}