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
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
【数据范围】
对于30%的数据,保证 N<=100
对于100%的数据,保证 N<=100000
对于所有的数据,保证 K<=100000,1<=X<=5,1<=A, B<=N
题解: 学习一下差分约束系统,学完之后就是裸题了。。
值得一提的是,最后加边的时候必须倒着加,要不然会挂的很惨。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#define N 100010
using namespace std;
int dis[N],l[N*10],n,k,kind,a,b,point[N],c[N],cnt,next[N*10];
long long ans;
struct use{int st,en,v;}e[N*10];
void add(int x,int y,int v){
next[++cnt]=point[x];point[x]=cnt;
e[cnt].st=x;e[cnt].en=y;e[cnt].v=v;
}
bool f[N],ff;
bool spfa(){
int h(0),t(1);
memset(dis,128,sizeof(dis));
l[t]=0;dis[0]=0;f[0]=true;
while (h<t){
int u=l[++h];f[u]=false;
for (int i=point[u];i;i=next[i]){
if (dis[e[i].en]<dis[u]+e[i].v)
{
dis[e[i].en]=dis[u]+e[i].v;
if (++c[e[i].en]>=n) return false;
if (!f[e[i].en]){l[++t]=e[i].en;f[e[i].en]=true;}
}
}
}
return true;
}
int main(){
scanf("%d%d",&n,&k);
for (int i=1;i<=k;i++){
scanf("%d%d%d",&kind,&a,&b);
if (kind==1){add(a,b,0);add(b,a,0);}
if (kind==2){if (a==b){cout<<"-1";return 0;}add(a,b,1);}
if (kind==3) add(b,a,0);
if (kind==4){if (a==b){cout<<"-1";return 0;}add(b,a,1);}
if (kind==5)add(a,b,0);
}
for(int i=n;i;--i) add(0,i,0);
if (!spfa()) {cout<<"-1";return 0;}
for (int i=1;i<=n;i++) {ans+=(long long)dis[i];if (dis[i]==0) ff=true;}
if (ff) ans+=(long long)n;
cout<<ans<<endl;
}