【bzoj2330】[SCOI2011]糖果

2015.04.14 16:49 Tue | 0次阅读 | 旧日oi | 固定链接 | 源码

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

【数据范围】
对于30%的数据,保证 N<=100
对于100%的数据,保证 N<=100000
对于所有的数据,保证 K<=100000,1<=X<=5,1<=A, B<=N

题解

差分约束,spfa求单源最长路,顺便拓扑判一下环
必须相等就互相连边权为0的边;不大于,不小于就从小的往大的连边权为0的边;大于,小于就从小的往大的连边权为1的边。

我的程序

#include<iostream>
#include<cstdio>
#define N 100005
using namespace std;
int n,k,cnt,q[N],head[N],cir[N],dis[N];
long long ans;
bool inq[N];
struct data{int to,next,v;}e[4*N];
void insert(int u,int v,int w){e[++cnt].to=v;e[cnt].next=head[u];e[cnt].v=w;head[u]=cnt;}
bool spfa()
{
     int t=0,w=1,now;
     q[0]=0;inq[0]=1;cir[0]=1;
     while(t!=w)
     {
               now=q[t];t++;if(t==N)t=0;
               for(int i=head[now];i;i=e[i].next)
               {
                       if(dis[now]+e[i].v>dis[e[i].to])
                       {
                           dis[e[i].to]=dis[now]+e[i].v;
                           if(++cir[e[i].to]>=n)return 0;
                           if(!inq[e[i].to])
                           {
                               inq[e[i].to]=1;q[w++]=e[i].to;
                               if(w==N)w=0;
                           }
                       }
               }
     inq[now]=0;
     }
     return 1;
 }
int main()
{
    scanf("%d%d",&n,&k);int x,a,b;
    while(k--)
    {
       scanf("%d%d%d",&x,&a,&b);
       switch(x)
       {
                case 1:insert(a,b,0);insert(b,a,0);break;
                case 2:if(a==b){printf("-1");return 0;}
                     insert(a,b,1);break;
                case 3:insert(b,a,0);break;
                case 4:if(a==b){printf("-1");return 0;}
                     insert(b,a,1);break;
                case 5:insert(a,b,0);break;
                }
       }
    for(int i=n;i>0;i--)insert(0,i,1);
    if(!spfa()){printf("-1");return 0;}
    for(int i=1;i<=n;i++)ans+=dis[i];
    printf("%lld",ans);
    return 0;
}```