BZOJ 2502: 清理雪道 | 有上下界最小流

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#define N 405
#define INF 0x3f3f3f3f
#define M 200005
using namespace std;
int head[N],cur[N],n,m,S,T,d[N],s,t,sum,ecnt=,lev[N],ans,dec;
queue<int> q;
struct adj
{
    int nxt,v,w;
}e[M];
void add(int u,int v,int w)
{
    e[++ecnt].v=v;e[ecnt].nxt=head[u];e[ecnt].w=w;head[u]=ecnt;
    e[++ecnt].v=u;e[ecnt].nxt=head[v];e[ecnt].w=;head[v]=ecnt;
}
bool bfs()
{
    for (int i=;i<=T;i++)
        cur[i]=head[i],lev[i]=-;
    q.push(S);lev[S]=;
    while (!q.empty())
    {
        int u=q.front();q.pop();
        for (int i=head[u],v;i;i=e[i].nxt)
            if (lev[v=e[i].v]==- && e[i].w>)
                q.push(v),lev[v]=lev[u]+;
    }
    return lev[T]!=-;
}
int dfs(int u,int flow)
{
    if (u==T) return flow;
    int ret=,v,delta;
    for (int &i=cur[u];i;i=e[i].nxt)
        if (lev[v=e[i].v]==lev[u]+ && e[i].w>)
        {
            delta=dfs(v,min(flow-ret,e[i].w));
            if (delta)
            {
                e[i].w-=delta;e[i^].w+=delta;ret+=delta;
                if (ret==flow) break;
            }
        }
    return ret;
}
int main()
{
    scanf("%d",&n);
    s=n+;t=n+;S=t+;T=S+;add(t,s,INF);
    for (int i=,u,v;i<=n;i++)
    {
        scanf("%d",&u);
        while (u--)
        {
            scanf("%d",&v);
            add(i,v,INF-);
            d[i]--,d[v]++;
        }
    }
    for (int i=;i<=n;i++)
        add(s,i,INF),add(i,t,INF);
    for (int i=;i<=n;i++)
        if (d[i]>) add(S,i,d[i]);
        else if (d[i]<) add(i,T,-d[i]);
    while (bfs()) ans+=dfs(S,INF);
    for (int i=head[S];i;i=e[i].nxt) e[i].w=e[i^].w=;
    for (int i=head[T];i;i=e[i].nxt) e[i].w=e[i^].w=;
    sum=e[].w;e[].w=e[].w=;
    S=t;T=s;
    while (bfs()) dec+=dfs(S,INF);
    printf("%d\n",sum-dec);
    return ;
}