P2756 飞行员配对方案问题 （dinic）

题目描述

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

对于给定的外籍飞行员与英国飞行员的配合情况，编程找出一个最佳飞行员配对方案，使皇家空军一次能派出最多的飞机。

思路

建立超级源点汇点跑dinic，记录匹配节点。

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N = 1e5 + 10;
int n, m, s, t, h[N], cnt, dep[N], cur[N], pre[N];
struct node {
    int v, w, nt;
} no[N];
void add(int u, int v, int w) {
    no[cnt] = node{v, w, h[u]};
    h[u] = cnt++;
}
int bfs() {
    queue<int> q;
    memset(dep, 0, sizeof dep);
    dep[s] = 1;
    q.push(s);
    while(!q.empty()) {
        int u = q.front();
        q.pop();
        for(int i = h[u]; ~i; i = no[i].nt) {
            int v = no[i].v;
            if(!dep[v] && no[i].w) {
                dep[v] = dep[u] + 1;
                q.push(v);
            }
        }
    }
    return dep[t] > 0;
}
int dfs(int u, int flow) {
    if(u == t)
        return flow;
    for(int i = cur[u]; ~i; i = no[i].nt) {
        int v = no[i].v;
        if(dep[v] == dep[u] + 1 && no[i].w) {
            int res = dfs(v, min(flow, no[i].w));
            if(res > 0) {
                no[i].w -= res;
                no[i ^ 1].w += res;
                if(u != s && v != t)
                    pre[v] = u;
                return res;
            }
        }
    }
    return 0;
}
int dinic() {
    int res = 0;
    while(bfs()) {
        for(int i = 0; i <= n; i++)
            cur[i] = h[i];
        while(int d = dfs(s, INF))
            res += d;
    }
    return res;
}
int main() {
    memset(h, -1, sizeof h);
    scanf("%d%d", &m, &n);
    int u, v;
    while(scanf("%d%d", &u, &v) && (u + v != -2))
        add(u, v, 1), add(v, u, 0);
    s = n + 1, t = n + 2;
    for(int i = 1; i <= m; i++)
        add(s, i, 1), add(i, s, 0);
    for(int i = m + 1; i <= n; i++)
        add(i, t, 1), add(t, i, 0);
    n = n + 2;
    int ans = dinic();
    if(!ans)
        return 0 * printf("No Solution!");
    printf("%d\n", ans);
    for(int i = m + 1; i <= n - 2; i++)
        if(pre[i])
            printf("%d %d\n", i, pre[i]);
    return 0;
}