BZOJ 2330: [SCOI2011]糖果

2330: [SCOI2011]糖果

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 5695  Solved: 1858
[Submit][Status][Discuss]

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最长路求最小值（求最大值时用最短路）

有注意到每个孩子必须有糖，所以在图外一点向每个点连一条权值为1的边，加和后即是答案

（数据中有环，答案会爆int（我干等着一个小时没改出来。。。。））

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iomanip>
using namespace std;
const int N=100010;
int n,m;
int dis[N];
struct EDGE{int nt,to,val;}edge[4*N];
int cnt,head[N];
void add(int u,int v,int val)
{edge[++cnt]=(EDGE){head[u],v,val};head[u]=cnt;}
int q[N],h,t,c[N];
bool inq[N];
bool spfa()
{
    h=1;t=2;q[1]=0;dis[0]=0;inq[0]=1;c[0]++;
    while(h!=t)
    {
        int now=q[h];h++;if(h==N)h=0;
        for(int i=head[now];i;i=edge[i].nt)
        {
            if(dis[edge[i].to]<dis[now]+edge[i].val)
            {
                dis[edge[i].to]=dis[now]+edge[i].val;
                if(++c[edge[i].to]>=n){cout<<-1;return 0;}
                if(!inq[edge[i].to])
                {q[t++]=edge[i].to;inq[edge[i].to]=1;if(t==N)t=0;}
            }
        }
        inq[now]=0;
    }
    return 1;
}
int main()
{
    scanf("%d%d",&n,&m);
    int opt,a,b;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&opt,&a,&b);
        switch(opt)
        {
            case 1:add(a,b,0);add(b,a,0);break;
            case 2:if(a==b){cout<<-1;return 0;}add(a,b,1);break;
            case 3:add(b,a,0);break;
            case 4:if(a==b){cout<<-1;return 0;}add(b,a,1);break;
            case 5:add(a,b,0);break;
        }
    }
    for(int i=n;i>0;i--)
    add(0,i,1);
    if(spfa())
    {
        long long ans=0;
        for(int i=1;i<=n;i++)
        ans+=dis[i];
        cout<<ans;
        return 0;
    }
}