教辅的组成

题目

https://www.luogu.org/problemnew/show/P1231

思路

这题和之前的网络流不同。

其实也差不多。为了避免“一书多用”，把书跟书多连一条边即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define maxn 1000077
using namespace std;
struct E
{
    int to,c,next;
}e[maxn];
int list[maxn],cnt=1,cur[maxn],dis[maxn],n1,n2,n3,m,x,y,s,t;
void add(int u,int v,int x)
{
    e[++cnt].to=v; e[cnt].c=x; e[cnt].next=list[u]; list[u]=cnt;
    e[++cnt].to=u; e[cnt].c=0; e[cnt].next=list[v]; list[v]=cnt;

}
bool bfs()
{
    memset(dis,0,sizeof(dis));
    queue<int> q;
    dis[0]=1; q.push(0);
    while(!q.empty())
    {
        int u=q.front(); q.pop();
        for(int i=list[u]; i; i=e[i].next)
        {
            if(e[i].c&&!dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+1;
                if(e[i].to==t) return 1;
                q.push(e[i].to);
            }
        }
    }
    return 0;
}
int dfs(int x,int maxf)
{
    if(x==t||!maxf) return maxf;
    int ret=0;
    for(int &i=cur[x]; i; i=e[i].next)
    {
        if(e[i].c&&dis[e[i].to]==dis[x]+1)
        {
            int f=dfs(e[i].to,min(maxf-ret,e[i].c));
            e[i].c-=f; e[i^1].c+=f;
            ret+=f;
            if(maxf==ret) break;
        }
    }
    return ret;
}
int dinic()
{
    int ass=0;
    while(bfs())
    {
        for(int i=s; i<=t; i++) cur[i]=list[i];
        ass+=dfs(s,0x3f3f3f3f);
    }
    return ass;
}
int main()
{
    scanf("%d%d%d",&n1,&n2,&n3);
    s=0; t=n1*2+n2+n3+1;
    for(int i=1; i<=n2; i++) add(s,i,1);
    scanf("%d",&m);
    for(int i=1; i<=m; i++)
    {
        scanf("%d%d",&x,&y);
        x+=n2; add(y,x,1);
    }
    for(int i=n2+1; i<=n1+n2; i++) add(i,n1+i,1);
    scanf("%d",&m);
    for(int i=1; i<=m; i++)
    {
        scanf("%d%d",&x,&y);
        x+=n1+n2; y+=n1*2+n2; add(x,y,1);
    }
    for(int i=n1*2+n2+1; i<=n1*2+n2+n3; i++) add(i,t,1);
    printf("%d",dinic());
}