1. 程式人生 > >BZOJ[2330]糖果 差分約束

BZOJ[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的邊,
如果ab就建一條由a指向b值為0的邊,
如果a=b就轉化為同時滿足b

aab,再像ab那樣建
然後再跑一遍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; }