【差分约束】BZOJ 2330: [SCOI2011]糖果

BZOJ 2330: [SCOI2011]糖果

---

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

---

Solution

差分约束裸题。。
建边 无解应该都很简单。。

Code

#include
#include
#include
#include
#include

#define maxn 100001

using namespace std;

struct ed{
    int last,to,c;
}edge[maxn*4];

int dis[maxn],inq[maxn],cir[maxn],tot=0,last[maxn],n;

void add(int u,int v,int c)
{
    edge[++tot].to=v,edge[tot].c=c,edge[tot].last=last[u],last[u]=tot;
}

bool spfa()
{
    dis[0]=1;
    queue<int>q;
    q.push(0);
    while(!q.empty())
    {
    int u=q.front();q.pop();
    if(++cir[u]>=n)return 0;
    inq[u]=0;
    for(int i=last[u];i;i=edge[i].last)
    {
        int v=edge[i].to;
        if(dis[u]+edge[i].c>dis[v])
        {
        dis[v]=dis[u]+edge[i].c;
        if(!inq[v])
        {
            inq[v]=1;
            q.push(v);
        }
        }
    }
    }
    return 1;
}

inline int in()
{
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return x;
}

int main()
{
    freopen("2330.in","r",stdin);
    int k,x,u,v;
    n=in(),k=in();
    for(int i=1;i<=k;i++)
    {
    x=in(),u=in(),v=in();
    if((x==4||x==2)&&u==v){printf("-1");return 0;}
    if(x==1)add(v,u,0),add(u,v,0);
    if(x==2)add(u,v,1);
    if(x==3)add(v,u,0);
    if(x==4)add(v,u,1);
    if(x==5)add(u,v,0);
    }
    for(int i=n;i>=0;i--)add(0,i,0);
    if(!spfa()){printf("-1");return 0;}
    long long an=0;
    for(int i=1;i<=n;i++)an+=dis[i];
    printf("%lld",an);
    return 0;
}