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的這篇部落格(強推)
如果
如果
如果
然後再跑一遍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;
}