[差分约束]poj 3169：Layout

大致题意：
    有n头牛，按照序号1……n排队，多头牛可以站在同一个位置(暂且这么认为)，也就是任意两头牛之间的距离都大于等于0。先给出ml组约束关系(u,v,w)代表第u牛和第v牛之间的距离要小于等于w。再给出md组关系(u,v,w)，代表第u牛和第v牛之间的距离要大于等于w。求这n头牛排成的队伍能否符合以上的约束，不能的话(也就是出现负环)，输出“-1”，如果距离是inf，输出“-2”。否则输出这个最短距离。

 

大致思路：

    转化为差分约束模型。dis[i]表示第i头牛所在的位置。要注意除了题目所给出的ml+md条约束外还要加入任意两头牛之间距离大于等于0的约束，也就是dis[i+1]>dis[i]。把1点设为最短路始点，如果spfa判断出负环的话就输出-1，如果dis[n]==inf则输出-2。否则输出dis[n]。

 

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
const int nMax=105000;
const int mMax=10000500;
const int inf=1<<28;
struct{
    int v, next;
    int  w;
}edge[mMax];
int n, k, head[nMax];
int dis[nMax];
int stack[nMax],m,sum[nMax];
bool vis[nMax];

void addedge(int a,int b,int w){
    edge[k].w = w;
    edge[k].v=b;
    edge[k].next=head[a];
    head[a]=k;k++;
}

bool spfa(int s){
    int i, top = 0;
    memset(vis,0,sizeof(vis));
    for(i=0;i<=n;i++)dis[i]=inf;
    dis[s]=0;
    stack[++top]=s;
    vis[s]=true;
    while(top){
        int u=stack[top--];
        for(i=head[u];i!=0;i=edge[i].next){
            int v=edge[i].v;
            if(dis[v]>dis[u]+edge[i].w){
                dis[v]=dis[u]+edge[i].w;
                if(!vis[v]){
                    vis[v]=true;
                    stack[++top] = v;
                    if(++sum[v]>n)return 0;
                }
            }
        }
        vis[u]=false;
    }
    return 1;
}
int main(){
    int m,ml,md,u,v,w,i;
    while(scanf("%d%d%d",&n,&ml,&md)!=EOF){
        k=1;
        memset(sum,0,sizeof(sum));
        memset(head,0,sizeof(head));
        while(ml--){
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
        }
        while(md--){
            scanf("%d%d%d",&u,&v,&w);
            addedge(v,u,-w);
        }
        for(i=1;i<=n;i++){
            addedge(i+1,i,0);
        }
        if(spfa(1)){
            if(dis[n]==inf){
                printf("-2\n");
            }
            else{
                printf("%d\n",dis[n]);
            }
        }
        else{
            printf("-1\n");
        }
    }
    return 0;
}