清理雪道

传送门

好题啊,由于每个点既可以进,也可以出,就可以新建一个源点和汇点,对于每个点都连边,然后就是最小流板子了。
代码:

#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
void read(int &x) {
    char ch; bool ok;
    for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
    for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
queue<int>q;int ans,ss,dis[1001],tt,n,m,s,t,inf=1e9,sum,pre[30001],cnt=1,in[301],nxt[30001],h[301],v[30001];
void add(int x,int y,int z)
{
    pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt,v[cnt]=z;
    pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt,v[cnt]=0;
}
bool bfs(int s,int t)
{
    memset(dis,0,sizeof dis);
    q.push(s);dis[s]=1;
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(rg int i=h[x];i;i=nxt[i])if(v[i]&&!dis[pre[i]])dis[pre[i]]=dis[x]+1,q.push(pre[i]);
    }
    return dis[t];
}
int dfs(int x,int flow,int t)
{
    if(x==t||!flow)return flow;
    int f=flow;
    for(rg int i=h[x];i;i=nxt[i])
        if(v[i]&&dis[pre[i]]>dis[x])
        {
            int y=dfs(pre[i],min(v[i],f),t);
            f-=y,v[i]-=y,v[i^1]+=y;
        }
    if(!f)dis[x]=-1;
    return flow-f;
}
int main()
{
    read(n);
    for(rg int i=1;i<=n;i++)
    {
        read(m);
        for(rg int j=1,x;j<=m;j++)read(x),add(i,x,inf-1),in[x]++,in[i]--;
    }
    ss=0,tt=n+1,s=n+2,t=n+3;
    for(int i=1;i<=n;i++)
    {
        if(in[i]>0)add(ss,i,in[i]);
        if(in[i]<0)add(i,tt,-in[i]);
        add(s,i,inf),add(i,t,inf);
    }
    add(t,s,inf);
    for(;bfs(ss,tt);sum+=dfs(ss,inf,tt));sum=v[cnt];
    for(int i=h[ss];i;i=nxt[i])v[i]=v[i^1]=0;
    for(int i=h[tt];i;i=nxt[i])v[i]=v[i^1]=0;
    v[cnt]=v[cnt^1]=0;
    for(;bfs(t,s);ans+=dfs(t,inf,s));
    printf("%d\n",sum-ans);
}

猜你喜欢

转载自www.cnblogs.com/lcxer/p/10229165.html
今日推荐