洛谷 P1231 教辅的组成

时间:2021-12-23 22:13:28
题意:

题目描述

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

输入输出格式

输入格式:

第一行包含三个正整数N1、N2、N3,分别表示书的个数、练习册的个数和答案的个数。

第二行包含一个正整数M1,表示书和练习册可能的对应关系个数。

接下来M1行每行包含两个正整数x、y,表示第x本书和第y本练习册可能对应。(1<=x<=N1,1<=y<=N2)

第M1+3行包含一个正整数M2,表述书和答案可能的对应关系个数。

接下来M2行每行包含两个正整数x、y,表示第x本书和第y本答案可能对应。(1<=x<=N1,1<=y<=N3)

输出格式:

输出包含一个正整数,表示最多可能组成完整书册的数目。

输入输出样例

输入样例#1:
5 3 45
4 3
2 2
5 2
5 1
5 3
5
1 3
3 1
2 2
3 3
4 3
输出样例#1:
2

说明

样例说明:

如题,N1=5,N2=3,N3=4,表示书有5本、练习册有3本、答案有4本。

M1=5,表示书和练习册共有5个可能的对应关系,分别为:书4和练习册3、书2和练习册2、书5和练习册2、书5和练习册1以及书5和练习册3。

M2=5,表示数和答案共有5个可能的对应关系,分别为:书1和答案3、书3和答案1、书2和答案2、书3和答案3以及书4和答案3。

所以,以上情况的话最多可以同时配成两个书册,分别为:书2+练习册2+答案2、书4+练习册3+答案3。

数据规模:

对于数据点1, 2, 3,M1,M2<= 20

对于数据点4~10,M1,M2 <= 20000


分析:将其转化为网络流最大流模型。连边,按照S->答案->书->册->T连容量为1的边。注意要把每本书拆为2个点,连一条容量为1的边,保证每本书只会被用到一次。


代码:

#include<cstdio>
#include<cstring>
using namespace std;

#define rep(i,a,b) for (int i=a;i<=b;++i)
#define to e[i].v
#define inf 1234567890

int head[2000005],h[2000005],q[2000005],flow,n1,n2,n3,m,id,S,T;

struct node{
    int next,v,w;
}e[3000005];

void add(int u,int v,int w) {e[++id].v=v; e[id].w=w; e[id].next=head[u]; head[u]=id;}
int max(int a,int b) {return a>b? a:b;}
int min(int a,int b) {return a>b? b:a;}
void swap(int &a,int &b) {int t=a;a=b;b=t;}

int read()
{
    int x=0,f=1;char ch=getchar();
    for (;ch>'9'||ch<'0';ch=getchar()) if (ch=='-') f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x*f;
}

void insert(int u,int v,int w) {add(u,v,w); add(v,u,0);}

int get11(int x) {return x;}
int get12(int x) {return x+n1;}
int get2(int x) {return n1*2+x;}
int get3(int x) {return n1*2+n2+x;}

int bfs()
{
    memset(h,-1,sizeof(h));
    int l,r;q[l=1]=0; h[q[1]]=0; r=1;
    for (;l<=r;)
    {
        int u=q[l++];
        for (int i=head[u];~i;i=e[i].next)
        if (e[i].w && h[to]<0)
        {
            h[to]=h[u]+1;
            q[++r]=to;
        }
    }
    return h[T]==-1? 0:1;
}

int dfs(int x,int in)
{
    int t,ans=0;
    if (x==T) return in;
    for (int i=head[x];~i;i=e[i].next)
    if (h[to]==h[x]+1 && e[i].w)
    {
        ans+=t=dfs(to,min(in-ans,e[i].w));
        e[i].w-=t; e[i^1].w+=t;
        if (ans==in) return in;
    }
    if (!ans) h[x]=-1;
    return ans;
}

void dinic()
{
    flow=0;
    while (bfs()) flow+=dfs(S,inf);
}

int main()
{
    memset(head,-1,sizeof(head));
    n1=read(); n2=read(); n3=read(); id=1; S=0; T=2000001;
    rep(i,1,n3) insert(S,get3(i),1);
    rep(i,1,n2) insert(get2(i),T,1);
    rep(i,1,n1) insert(get11(i),get12(i),1);
    m=read();
    rep(i,1,m)
    {
        int u=read(),v=read();
        insert(get12(u),get2(v),1);
    }
    m=read();
    rep(i,1,m)
    {
        int u=read(),v=read();
        insert(get3(v),get11(u),1);
    }
    dinic();
    printf("%d",flow);
    return 0;
}