洛谷 P1231 教辅的组成（最大流）

任重而道远

题目描述

蒟蒻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）

输出格式：

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

AC代码：

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

const int N = 1e6 + 10;
const int INF = 1e9 + 7;
struct edge {
    int tov, nxt, flw;
}e[N];
int head[N], dis[N];
int n1, n2, n3, m1, m2, num = 1, sink, src, ans = 0;
bool vis[N];

void add_edge ( int u, int v, int f ) {
    num++;
    e[num].tov = v;
    e[num].nxt = head[u];
    e[num].flw = f;
    head[u] = num;
    num++;
    e[num].tov = u;
    e[num].nxt = head[v];
    e[num].flw = 0;
    head[v] = num;
}

bool bfs () {
    memset (vis, false, sizeof (vis));
    memset (dis, 0, sizeof (dis));
    queue <int> q;
    q.push (src);
    vis[src] = true;
    while (!q.empty ()) {
        int u = q.front ();
        q.pop ();
        for (int i = head[u]; i; i = e[i].nxt) {
            int v = e[i].tov;
            if (e[i].flw && !vis[v]) {
                q.push (v);
                dis[v] = dis[u] + 1;
                vis[v] = true;
            }
        }
    }
    return vis[sink];
}

int dfs ( int u, int delta ) {
    if (u == sink) return delta;
    int res = 0;
    for (int i = head[u]; i && delta; i = e[i].nxt) {
        int v = e[i].tov;
        if (e[i].flw && dis[v] == dis[u] + 1) {
            int dd = dfs (v, min (delta, e[i].flw));
            e[i].flw -= dd;
            e[i ^ 1].flw += dd;
            delta -= dd;
            res += dd;
        }
    }
    return res;
}

int main () {
    scanf ("%d%d%d", &n1, &n2, &n3);
    src = 0; sink = n1 * 2 + n2 + n3 + 1;
    for (int i = 1; i <= n2; i++)
      add_edge (src, i, 1);
    for (int i = 2 * n1 + n2 + 1; i <= 2 * n1 + n2 + n3; i++)
      add_edge (i, sink, 1);
    scanf ("%d", &m1);
    for (int i = 1; i <= m1; i++) {
    	int x, y;
        scanf ("%d%d", &x, &y);
        add_edge (y, n2 + x, 1);
    }
    for (int i = n2 + 1; i <= n2 + n1; i++)
      add_edge (i, i + n1, 1);
    scanf ("%d", &m2);
    for (int i = 1; i <= m2; i++) {
    	int x, y;
    	scanf ("%d%d", &x, &y);
    	add_edge (n2 + n1 + x, 2 * n1 + n2 + y, 1);
    }
    while (bfs ())
      ans += dfs (src, INF);
    printf ("%d", ans);
    return 0;
}