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

Posted 2020-10-20 MSPqwq

 

 

 

#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=1,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=0;head[v]=ecnt;
}
bool bfs()
{
    for (int i=1;i<=T;i++)
        cur[i]=head[i],lev[i]=-1;
    q.push(S);lev[S]=1;
    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]==-1 && e[i].w>0)
                q.push(v),lev[v]=lev[u]+1;
    }
    return lev[T]!=-1;
}
int dfs(int u,int flow)
{
    if (u==T) return flow;
    int ret=0,v,delta;
    for (int &i=cur[u];i;i=e[i].nxt)
        if (lev[v=e[i].v]==lev[u]+1 && e[i].w>0)
        {
            delta=dfs(v,min(flow-ret,e[i].w));
            if (delta)
            {
                e[i].w-=delta;e[i^1].w+=delta;ret+=delta;
                if (ret==flow) break;
            }
        }
    return ret;
}
int main()
{
    scanf("%d",&n);
    s=n+1;t=n+2;S=t+1;T=S+1;add(t,s,INF);
    for (int i=1,u,v;i<=n;i++)
    {
        scanf("%d",&u);
        while (u--)
        {
            scanf("%d",&v);
            add(i,v,INF-1);
            d[i]--,d[v]++;
        }
    }
    for (int i=1;i<=n;i++)
        add(s,i,INF),add(i,t,INF);
    for (int i=1;i<=n;i++)
        if (d[i]>0) add(S,i,d[i]);
        else if (d[i]<0) 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^1].w=0;
    for (int i=head[T];i;i=e[i].nxt) e[i].w=e[i^1].w=0;
    sum=e[3].w;e[2].w=e[3].w=0;
    S=t;T=s;
    while (bfs()) dec+=dfs(S,INF);
    printf("%d\n",sum-dec);
    return 0;
}