BZOJ 2330 [SCOI2011] 糖果 (差分约束)

https://www.lydsy.com/JudgeOnline/problem.php?id=2330
题意：幼儿园里有$N$个小朋友，老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。小朋友们会提出$K$个要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时候，要求的具体形式如下：

幼儿园的糖果总是有限的，老师想知道他至少需要准备多少个糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。

很明显的一道差分约束题目，对于不同要求，进行相应连边，连边方式如下：
$X=1:\ A=B:\ addedge(a,b,0)\ addedge(b,a,0)$
$X=2:\ A<B:\ addedge(a,b,1)$
$X=3:\ A≥B:\ addedge(b,a,0)$
$X=4:\ A>B:\ addedge(b,a,1)$
$X=5:\ A≤B:\ addedge(a,b,0)$
因为这里既有大于等于的情况也有小于等于的情况，为了避免连负权值的边，这里相当于反向连边然后跑最长路。因为要求每个小朋友都要分到糖果，对于第$i$个点，我们$addedge(0,i,1)$，相当于加了一个超级起点，然后令这个起点的权值为0，每个小朋友都要比这个起点多至少1，即可保证每个小朋友得到至少1个糖果。然后用SPFA跑一边最长路，就可以得到从超级起点0到每一个小朋友最少需要$dis[i]$个糖果，最后的答案就是$\sum dis[i]$。

差分约束的题跑SPFA的时候貌似都要用stack存状态，用queue往往会T掉。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<cstring>
#define INF 0x3f3f3f
using namespace std;

const int maxn=1000100;
int head[maxn],k=1,n,K,x,y,a,b,dis[maxn],sum[maxn];
bool vis[maxn],v[maxn];
struct edge{
    int w,v,next;
}e[maxn*2];
void addedge(int u,int v,int w){e[k].v=v;e[k].w=w;e[k].next=head[u];head[u]=k++;}
long long spfa(int s)
{
    long long res=0;
    memset(vis,0,sizeof(vis));
    memset(dis,-1,sizeof(dis));
    stack<int> q;
    dis[s]=0;
    vis[s]=1;
    q.push(s);
    while(!q.empty())
    {
        int t=q.top();
        vis[t]=0;
        q.pop();
        for(int i=head[t];i;i=e[i].next)
        {
            int v=e[i].v;
            if(dis[v]<dis[t]+e[i].w)
            {
                dis[v]=dis[t]+e[i].w;
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);
                    sum[v]++;
                    if(sum[v]>n)
                        return -1;
                }
            }
        }
    }
    for(int i=1;i<=n;i++)
        res+=dis[i];
    return res;
}

int main()
{
    scanf("%d%d",&n,&K);
    for(int i=1;i<=K;i++)
    {
        scanf("%d%d%d",&x,&a,&b);
        v[a]=1;v[b]=1;
        switch(x)
        {
        case 1:
            addedge(a,b,0);addedge(b,a,0);break;
        case 2:
            addedge(a,b,1);break;
        case 3:
            addedge(b,a,0);break;
        case 4:
            addedge(b,a,1);break;
        case 5:
            addedge(a,b,0);break;
        }
    }
    for(int i=1;i<=n;i++)
        addedge(0,i,1);
    long long ans;
    ans=spfa(0);
    printf("%lld\n",ans);
    return 0;
}