【ACWing】2175. 飞行员配对方案问题

题目地址：

https://www.acwing.com/problem/content/2177/

第二次世界大战时期，英国皇家空军从沦陷国征募了大量外籍飞行员。由皇家空军派出的每一架飞机都需要配备在航行技能和语言上能互相配合的$2$名飞行员，其中$1$名是英国飞行员，另$1$名是外籍飞行员。在众多的飞行员中，每一名外籍飞行员都可以与其他若干名英国飞行员很好地配合。如何选择配对飞行的飞行员才能使一次派出最多的飞机。对于给定的外籍飞行员与英国飞行员的配合情况，试设计一个算法找出最佳飞行员配对方案，使皇家空军一次能派出最多的飞机。

输入格式：
第$1$行有$2$个正整数$m$和$n$。$m$是外籍飞行员数；$n$是皇家空军的飞行员总数。外籍飞行员编号为$1\sim m$；英国飞行员编号为$m+1\sim n$。接下来每行有$2$个正整数$i$和$j$，表示外籍飞行员$i$可以和英国飞行员$j$配合。文件最后以$2$个$-1$结束。

输出格式：
第$1$行是最佳飞行员配对方案一次能派出的最多的飞机数$M$。接下来$M$行是最佳飞行员配对方案。每行有$2$个正整数$i$和$j$，表示在最佳飞行员配对方案中，外籍飞行员$i$和英国飞行员$j$配对。如果有多种配对方案，则输出任意一种即可，方案内部配对输出顺序随意。

数据范围：
$1<m<n<100$

其实是二分图匹配问题，可以用最大流的Dinic算法来做。建图是这样的，想象最左边有个源点$S$，最右边有个汇点$T$，$S$向左部每个点连一条容量是$1$的有向边，然后右部每个点向$T$连一条容量是$1$的有向边，对于中间的点，左部的点和右部的点如果能配合则连一条有向边，容量也是$1$。首先容易知道任何一个匹配都对应着一条可行流；接着，也容易看出任何一个整数流都对应着一个匹配，而Dinic算法无论是找增广路还是更新残留网络，所得结果都是整数，所以Dinic算法找到的最大流也是整数流，所以Dinic算法能找到最大匹配。找到最大流之后，只需找一下残留网络中间的从左向右的边里哪些边的容量是$0$即可，容量是$0$意味着原图里有流量$1$，即有一个匹配。代码如下：

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

const int N = 110, M = 5210, INF = 1e8;
int m, n, S, T;
int h[N], e[M], f[M], ne[M], idx;
int q[N], d[N], cur[N];

void add(int a, int b, int c) {
    e[idx] = b, ne[idx] = h[a], f[idx] = c, h[a] = idx++;
    e[idx] = a, ne[idx] = h[b], f[idx] = 0, h[b] = idx++;
}

bool bfs() {
    int hh = 0, tt = 0;
    memset(d, -1, sizeof d);
    q[tt++] = S, d[S] = 0, cur[S] = h[S];
    while (hh < tt) {
        int t = q[hh++];
        for (int i = h[t]; ~i; i = ne[i]) {
            int v = e[i];
            if (d[v] == -1 && f[i]) {
                d[v] = d[t] + 1;
                if (v == T) return true;

                cur[v] = h[v];
                q[tt++] = v;
            }
        }
    }
    
    return false;
}

int dfs(int u, int limit) {
    if (u == T) return limit;
    int flow = 0;
    for (int i = cur[u]; ~i && flow < limit; i = ne[i]) {
        cur[u] = i;
        int v = e[i];
        if (d[v] == d[u] + 1 && f[i]) {
            int t = dfs(v, min(limit - flow, f[i]));
            if (!t) d[v] = -1;
            f[i] -= t, f[i ^ 1] += t, flow += t;
        }
    }

    return flow;
}

int dinic() {
    int r = 0, flow;
    while (bfs()) while(flow = dfs(S, INF)) r += flow;
    return r;
}

int main() {
    scanf("%d%d", &m, &n);
    S = 0, T = n + 1;
    memset(h, -1, sizeof h);

	// 从S向左部点连容量为1的边
    for (int i = 1; i <= m; i++) add(S, i, 1);
    // 从右部点向T连容量为1的边
    for (int i = m + 1; i <= n; i++) add(i, T, 1);

    int a, b;
    while (cin >> a >> b, ~a) add(a, b, 1);

    printf("%d\n", dinic());

	// 枚举从左向右的边（即正向边）
    for (int i = 0; i < idx; i += 2) 
    	// 右边点的编号范围是m + 1到n，找容量是0的边
        if (e[i] > m && e[i] <= n && !f[i]) 
            printf("%d %d\n", e[i ^ 1], e[i]);

    return 0;
}

时间复杂度$O(m\sqrt{n})$，空间$O(n)$。