bzoj2330 [SCOI2011]糖果 差分约束

这道题应该是一道比较裸的差分约束系统。

按照题意建图:

1、如果A和B一样多 -> 连边(A,B,0)，(B,A,0)

2、如果A小于B -> 连边(A,B,1)

3、如果A大于等于B -> 连边(B,A,0)

4、如果A大于B -> 连边(B,A,1)

5、如果A小于等于B -> (A,B,0)

其中第2种情况和第4种情况要特判，若a == b则无解。

这样建好后用spfa求一次最长路就可以了。

为什么要最长路呢，因为要求的是总长度而不是一个点的长度。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
using namespace std;
#define maxn 210000
int n,k,num;
int b[maxn],val[maxn],in[maxn],next[maxn],p[maxn],dis[maxn];
bool vis[maxn];
void add(int x,int y,int z)
{
    num++;
    b[num]=y;
    val[num]=z;
    next[num]=p[x];
    p[x]=num;
}
queue<int>q;
bool spfa()
{
    int tot,cur;
    while(!q.empty())
    {
        cur=q.front();
        q.pop();
        vis[cur]=false;
        for(tot=p[cur];tot;tot=next[tot])
        {
            if(dis[cur]+val[tot]>dis[b[tot]])
            {
                dis[b[tot]]=dis[cur]+val[tot];
                in[tot]++;
                if(in[tot]>n) return false;
                if(!vis[b[tot]])
                {
                    vis[b[tot]]=1;
                    q.push(b[tot]);
                }
            }
        }
    }
    return true;
}
int main()
{
    int u,v,op;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=k;i++)
    {
        scanf("%d%d%d",&op,&u,&v);
        if(op==1)
        {
            add(u,v,0);
            add(v,u,0);
        }
        if(op==2)
        {
            add(u,v,1);
            if(u==v)
            {
                printf("-1\n");
                return 0;
            }
        }
        if(op==3)
        {
            add(v,u,0);
        }
        if(op==4)
        {
            add(v,u,1);
            if(u==v)
            {
                printf("-1\n");
                return 0;
            }
        }
        if(op==5)
        {
            add(u,v,0);
        }
    }
    for(int i=1;i<=n;i++)
    {
        vis[i]=1;
        dis[i]=1;
        q.push(i);
    }
    if(!spfa())
    {
        printf("-1\n");
        return 0;
    }

    long long tmp=0;
    for(int i=1;i<=n;i++)
    {
        tmp+=dis[i];
    }
    printf("%lld\n",tmp);
    return 0;
}