糖果_bzoj2330_差分约束系统

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。

数据范围

---

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

Analysis

---

看到分类是差分约束系统就顺手刷了啊啊啊
五种关系分别可以表示成

1. $A-B\leq 0$ 且 $A-B\geq 0$
2. $B-A\leq -1$
3. $B-A\leq 0$
4. $B-A\leq -1$
5. $A-B\leq 0$

连边，然后用一个哲♂学点作为源点链接所有其他点，这一步要倒着加边因为数据会有很大的正权环
至于特判正环什么的spfa就不说了

Code

---

#include <stdio.h>
#include <string.h>
#include <queue>
#define ll long long
#define N 100005
#define M 300005
using namespace std;
struct edge{int x,y,w,next;}e[M+1];
int dis[N+1],cnt[N+1],ls[M+1],n,m,maxE=0;
bool vis[N+1];
void add(int x,int y,int w)
{
    e[++maxE]=(edge){x,y,w,ls[x]};
    ls[x]=maxE;
}
ll spfa(int st)
{
    queue<int>q;
    q.push(st);
    vis[st]=true;
    dis[st]=0;
    for (int i=1;i<=maxE;i++)
        if (e[i].x==e[i].y&&e[i].w)
            return -1;
    while (!q.empty())
    {
        int now=q.front();q.pop();
        for (int i=ls[now];i;i=e[i].next)
            if (e[i].w+dis[now]>dis[e[i].y])
            {
                if (++cnt[e[i].y]>=n)
                    return -1;
                dis[e[i].y]=dis[now]+e[i].w;
                if (!vis[e[i].y])
                {
                    vis[e[i].y]=true;
                    q.push(e[i].y);
                }
            }
        vis[now]=false;
    }
    ll tot=0;
    for (int i=1;i<=n;i++)
        tot+=dis[i];
    return tot;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    {
        int t,x,y;
        scanf("%d%d%d",&t,&x,&y);
        switch (t)
        {
            case 1:
            {
                add(x,y,0);
                add(y,x,0);
                break;
            }
            case 2:
                add(x,y,1);
                break;
            case 3:
                add(y,x,0);
                break;
            case 4:
                add(y,x,1);
                break;
            case 5:
                add(x,y,0);
                break;
        }
    }
    for (int i=n;i>=1;i--)
        add(0,i,1);
    printf("%lld\n",spfa(0));
    return 0;
}