bzoj2330 [SCOI2011]糖果

2330: [SCOI2011]糖果

Time Limit: 10 Sec Memory Limit: 128 MB

Description

幼儿园里有N个小朋友，lxhgww老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。但是小朋友们也有嫉妒心，总是会提出一些要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时候，lxhgww需要满足小朋友们的K个要求。幼儿园的糖果总是有限的，lxhgww想知道他至少需要准备多少个糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。

Input

输入的第一行是两个整数N，K。

接下来K行，表示这些点需要满足的关系，每行3个数字，X，A，B。

如果X=1，表示第A个小朋友分到的糖果必须和第B个小朋友分到的糖果一样多；

如果X=2，表示第A个小朋友分到的糖果必须少于第B个小朋友分到的糖果；

如果X=3，表示第A个小朋友分到的糖果必须不少于第B个小朋友分到的糖果；

如果X=4，表示第A个小朋友分到的糖果必须多于第B个小朋友分到的糖果；

如果X=5，表示第A个小朋友分到的糖果必须不多于第B个小朋友分到的糖果；

Output

输出一行，表示lxhgww老师至少需要准备的糖果数，如果不能满足小朋友们的所有要求，就输出-1。

Sample Input

5 7

1 1 2

2 3 2

4 4 1

3 4 5

5 4 5

2 3 5

4 5 1

Sample Output

HINT

【数据范围】

对于30%的数据，保证 N<=100

对于100%的数据，保证 N<=100000

对于所有的数据，保证 K<=100000，1<=X<=5，1<=A, B<=N

Tip：

　　此题是差分约束，不知道的同学自寻百度；

　　若x=1，A=B --> A+0<=B， B+0<=A，连（A，B，0）（B，A，0）;

　　若x=2，A<B --> A+1<=B，连（A，B，1）;

　　若x=3，A>=B --> B+0<=A，连（B，A，0）;

　　若x=4，A>B --> B+1<=A，连（B，A，1）;

　　若x=5，A<=B --> A+0<=B，连（A，B，0）;

　　连（0，i，1）；

　　最后从第0号点做一次最长路；

Code：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define MAXN 1000005
using namespace std;

int n,m,head[MAXN],vet[MAXN],next[MAXN],len[MAXN],d[MAXN];
int qu[MAXN+1],cir[MAXN],st,ed,tot;
bool inq[MAXN];
long long ans;

void add(int x,int y,int z){tot++;next[tot]=head[x];head[x]=tot;vet[tot]=y;len[tot]=z;}

bool spfa(){
    int x,y;
    st=0; ed=0;
    qu[++ed]=0;inq[0]=1; cir[0]=1;
    while(st!=ed){
        x=qu[++st];
        if(st==MAXN) st=0;
        for(int i=head[x];i;i=next[i]){
            y=vet[i];
            if(d[x]+len[i]>d[y]){
                d[y]=d[x]+len[i];
                if(++cir[y]>=n) return false;
                if(!inq[y]){
                    inq[y]=1;
                    qu[++ed]=y;
                    if(ed==MAXN) ed=0;
                }
            }
        }
        inq[x]=0;
    }
    return true;
}

int main(){
    memset(cir,0,sizeof(cir));
    tot=0;
    scanf("%d%d",&n,&m);
    int x,y,z;
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&z,&x,&y);
        if(z==1){
            add(x,y,0);
            add(y,x,0);
        }
        if(z==2){
            if(x==y){
                printf("-1");
                return 0;
            }
            add(x,y,1);
        }
        if(z==3){
            add(y,x,0);
        }
        if(z==4){
            if(x==y){
                printf("-1");
                return 0;
            }
            add(y,x,1);
        }
        if(z==5){
            add(x,y,0);
        }
    }
    for(int i=n;i>0;i--) add(0,i,1);
    if(!spfa()){
        printf("-1");
        return 0;
    }
    for(int i=1;i<=n;i++){
        ans+=d[i];
    }
    printf("%lld",ans);
    return 0;
}