[SCOI2011]糖果

題目描述

幼兒園裏有 N個小朋友，lxhgww老師現在想要給這些小朋友們分配糖果，要求每個小朋友都要分到糖果。但是小朋友們也有嫉妒心，總是會提出一些要求，比如小明不希望小紅分到的糖果比他的多，於是在分配糖果的時候，lxhgww 需要滿足小朋友們的 K個要求。幼兒園的糖果總是有限的，lxhgww 想知道他至少需要準備多少個糖果，才能使得每個小朋友都能夠分到糖果，並且滿足小朋友們所有的要求。

輸入格式

輸入的第一行是兩個整數 N，K。接下來 KK 行，表示這些點需要滿足的關係，每行 3 個數字，X，A，B。

如果 X=1， 表示第 A 個小朋友分到的糖果必須和第 B 個小朋友分到的糖果一樣多；

如果 X=2， 表示第 A個小朋友分到的糖果必須少於第 B 個小朋友分到的糖果；

如果 X=3， 表示第 A個小朋友分到的糖果必須不少於第 B 個小朋友分到的糖果；

如果 X=4， 表示第 A個小朋友分到的糖果必須多於第 B個小朋友分到的糖果；

如果 X=5， 表示第 A個小朋友分到的糖果必須不多於第 B個小朋友分到的糖果；

輸出格式

輸出一行，表示 lxhgww 老師至少需要準備的糖果數，如果不能滿足小朋友們的所有要求，就輸出 -1。

---

差分約束裸題。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define maxn 100001
#define maxm 100001
using namespace std;
int n,m;
long long ans;
inline int read(){
    register int x(0),f(1); register char c(getchar());
    while(c<'0'||'9'<c){ if(c=='-') f=-1; c=getchar(); }
    while('0'<=c&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x*f;
}

struct edge{
    int to,dis,next;
    edge(){}
    edge(const int &_to,const int &_dis,const int &_next){ to=_to,dis=_dis,next=_next; }
}e[maxm<<2];
int head[maxn],k;
inline void add(const int &u,const int &v,const int &w){ e[k]=edge(v,w,head[u]),head[u]=k++; }

long long dis[maxn];
int cnt[maxn];
bool inq[maxn];
queue<int> q;
inline bool spfa(){
    q.push(0),inq[0]=true;
    while(q.size()){
        int u=q.front(); inq[u]=false,q.pop();
        for(register int i=head[u];~i;i=e[i].next){
            int v=e[i].to;
            if(dis[v]<dis[u]+e[i].dis){
                dis[v]=dis[u]+e[i].dis,cnt[v]=cnt[u]+1;
                if(cnt[v]==n+1) return false;
                if(!inq[v]) q.push(v),inq[v]=true;
            }
        }
    }
    return true;
}

int main(){
    memset(head,-1,sizeof head);
    n=read(),m=read();
    for(register int i=1;i<=m;i++){
        int op=read(),u=read(),v=read();
        if(op==1) add(u,v,0),add(v,u,0);
        if(op==2){
            if(u==v){ puts("-1"); return 0; }
            add(u,v,1);
        }
        if(op==3) add(v,u,0);
        if(op==4){
            if(u==v){ puts("-1"); return 0; }
            add(v,u,1);
        }
        if(op==5) add(u,v,0);
    }
    for(register int i=n;i>=1;i--) add(0,i,1);
    if(!spfa()) puts("-1");
    else{
        for(register int i=1;i<=n;i++) ans+=dis[i];
        printf("%lld\n",ans);
    }
    return 0;
}