poj3169 Layout

Layout

题意：n头牛编号为1到n，按照编号的顺序排成一列，每两头牛的之间的距离 >= 0。这些牛的距离存在着一些约束关系：1.有ml组（u, v, w）的约束关系，表示牛[u]和牛[v]之间的距离必须 <= w。2.有md组（u, v, w）的约束关系，表示牛[u]和牛[v]之间的距离必须 >= w。问如果这n头无法排成队伍，则输出-1，如果牛[1]和牛[n]的距离可以无限远，则输出-2，否则则输出牛[1]和牛[n]之间的最大距离。

代码：

#include<iostream>
#include<string>
#include<queue>
#include<map>
#include<cmath>
#include<stack>
#include<algorithm>
using namespace std;
// 队列实现，而且有负权回路判断—POJ 3169 Layout

const int INF = 0x3F3F3F3F;
const int V = 1001;
const int E = 20001;
int pnt[E], cost[E], nxt[E];
int e, head[V], dist[V];
bool vis[V];
int cnt[V]; // 入队列次数
queue<int> que;

void init(){
    e = 0;
    memset(head, -1, sizeof(head));
}

inline void addedge(int u, int v, int c){
    pnt[e] = v; cost[e] = c; nxt[e] = head[u]; head[u] = e++;
}

int SPFA(int src, int n){
    int i,u,v;

    for( i=1; i <= n; ++i ) {
        dist[i] = INF;cnt[i]=0;vis[i]=1;
    }
    
    while(!que.empty())que.pop();
    que.push(src);vis[src] =0; 
    dist[src] = 0;
    ++cnt[src];
    
    while( !que.empty() ){
        u = que.front();que.pop(); vis[u] =1;
        for( i=head[u]; i != -1; i=nxt[i] ){
            v = pnt[i];
            if(dist[v] > dist[u] + cost[i] ){
                dist[v] = dist[u] + cost[i]; 
                if(vis[v]){
                    que.push(v);vis[v]=0;
                    if((++cnt[v])>n)return -1;//出现负权回路 
                }
            }
        }
    }
    if( dist[n] == INF ) return -2; // src与n不可达
    return dist[n]; // 返回src到n的最短距离
}


int main(){
    int n, ml, md;
    while( scanf("%d%d%d", &n, &ml, &md) != EOF ){
        int a, b, c;
        init();
        while(ml--){
            scanf("%d%d%d", &a, &b, &c);
            if( a > b) swap(a, b);
            addedge(a, b, c);
        }
        while(md--){
            scanf("%d%d%d", &a, &b, &c);
            if( a < b ) swap(a, b);
            addedge(a, b, -c);
        }

        printf("%d\n", SPFA(1, n));
    }
    return 0;
}