【LuoguP1231】教辅的组成

题目链接

题目描述

蒟蒻HansBug在一本语文书里面发现了一本答案，然而他却明明记得这书应该还包含一份练习题。然而出现在他眼前的书多得数不胜数，其中有书，有答案，有练习册。已知一个完整的书册均应该包含且仅包含一本书、一本练习册和一份答案，然而现在全都乱做了一团。许多书上面的字迹都已经模糊了，然而HansBug还是可以大致判断这是一本书还是练习册或答案，并且能够大致知道一本书和答案以及一本书和练习册的对应关系（即仅仅知道某书和某答案、某书和某练习册有可能相对应，除此以外的均不可能对应）。既然如此，HansBug想知道在这样的情况下，最多可能同时组合成多少个完整的书册。

题解

实质是个三分图匹配(自己YY的) ,一道网络流模板题。
和二分图匹配类似，但中间的图要拆成两个点，并连上容量为1的边，保证中间的点不被匹配多次。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#define Set(a,b) memset(a,b,sizeof(a))
#define Copy(a,b) memcpy(a,b,sizeof(a))
using namespace std;
inline int read()
{
    int x=0;char ch=getchar();int t=1;
    for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') t=-1;
    for(;ch>='0'&&ch<='9';ch=getchar()) x=(x<<1)+(x<<3)+(ch-48);
    return x*t;
}
const int M=1e6+20;
const int INF=1e9;
const int N=5e4;
struct edge{
    int to,next,cap;
}a[M<<1];
int head[N];
int dis[N];
int cnt=0;
int cur[N];
int n,m,k;
int tot;
inline void add(int x,int y,int z)
{
    a[cnt]=(edge){y,head[x],z};
    head[x]=cnt++;
}
queue<int> Q;
inline void clear(queue<int> &Q)
{
    queue<int> p;
    swap(p,Q);
}
inline bool bfs()
{
    clear(Q);
    Set(dis,-1);
    Q.push(0);dis[0]=0;
    while(!Q.empty())
    {
        register int u=Q.front();Q.pop();
        for(register int v,i=head[u];i!=-1;i=a[i].next){
            v=a[i].to;
            if(dis[v]!=-1||a[i].cap==0) continue;
            dis[v]=dis[u]+1;
            if(v==tot) return 1;
            Q.push(v);
        }
    }
    return (dis[tot]!=-1);
}
inline int dfs(int u,int flow)
{
    if(u==tot) return flow;
    //printf("%d\n",u);
    register int delta=flow;
    for(register int v,&i=cur[u];i!=-1;i=a[i].next)
    {
        v=a[i].to;
        if(a[i].cap==0||(dis[v]!=dis[u]+1)) continue;
        register int d=dfs(v,min(delta,a[i].cap));
        delta-=d;
        a[i].cap-=d;a[i^1].cap+=d;
        if(d==0) dis[v]=0;
        if(delta==0) break;
    }
    return flow-delta;
}
inline void Dinic()
{
    register int max_flow=0;
    while(bfs()) Copy(cur,head),max_flow+=dfs(0,INF);
    printf("%d\n",max_flow);
}
int main()
{
    n=read();m=read();k=read();
    register int jj=2*n;
    tot=jj+m+k+1;
    register int op=read();Set(head,-1);
    for(register int i=1;i<=m;i++){add(0,jj+i,1);add(jj+i,0,0);}
    for(register int i=1;i<=k;i++){add(jj+m+i,tot,1);add(tot,jj+m+i,0);}
    for(register int i=1;i<=n;i++) add(i,i+n,1),add(i+n,i,0);
    register int x,y;
    for(register int i=1;i<=op;i++){
        x=read();y=read();
        add(jj+y,x,1);add(x,jj+y,0);
    }
    op=read();
    for(register int i=1;i<=op;i++){
        x=read();y=read();add(x+n,jj+m+y,1);add(jj+m+y,x+n,0);
    }
    Dinic();
}