BZOJ[2330]糖果 差分约束

题目链接http://www.lydsy.com/JudgeOnline/problem.php?id=2330

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
11

HINT

【数据范围】

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

差分约束裸题..不会差分约束的同学可以看看void_man的这篇博客(强推)

如果$a < b$就建一条由a指向b值为1的边，
如果$a \le b$就建一条由a指向b值为0的边，
如果$a = b$就转化为同时满足$b \le a$和$a \le b$，再像$a \le b$那样建
然后再跑一遍spfa就可以了

代码如下：

#include<algorithm>
#include<cstdio>
#include<queue>
#define N 1000020
using namespace std;
queue<int>q;
int n,m,k,x,y,top;
long long ans;
int fir[N],dis[N],num[N];
bool b[N];
void GG(){
    printf("-1");
    exit(0);
}
struct Edge{
    int to,nex,k;
    Edge(){}
    Edge(int _,int __,int ___):to(_),nex(__),k(___){}
}nex[N*2];
void add(int x,int y,int k){
    nex[++top]=Edge(y,fir[x],k);
    fir[x]=top;
}
bool spfa(){
    while(!q.empty()) q.pop();
    num[0]=1;dis[0]=0;q.push(0);
    while(!q.empty()){
        int x=q.front();q.pop();
        b[x]=false;
        for(int i=fir[x];i;i=nex[i].nex){
            if(dis[nex[i].to]<dis[x]+nex[i].k){
                dis[nex[i].to]=dis[x]+nex[i].k;
                if(++num[nex[i].to]>=n) return false;
                if(!b[nex[i].to]){
                    b[nex[i].to]=true;
                    q.push(nex[i].to);
                }
            }
        }
    }
    return true;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&k,&x,&y);
        switch(k){
            case 1:add(x,y,0);add(y,x,0);break;
            case 2:if(x==y) GG();add(x,y,1);break;
            case 3:add(y,x,0);break;
            case 4:if(x==y) GG();add(y,x,1);break;
            case 5:add(x,y,0);break;
        }
    }
    for(int i=n;i;i--) add(0,i,1);
    if(!spfa()) GG();
    for(int i=1;i<=n;i++) ans=ans+dis[i];
    printf("%lld",ans);
return 0;
}