P2756 飞行员配对方案问题【网络流+二分图匹配】

题目描述
英国皇家空军从沦陷国征募了大量外籍飞行员。由皇家空军派出的每一架飞机都需要配备在航行技能和语言上能互相配合的2 名飞行员,其中1 名是英国飞行员,另1名是外籍飞行员。在众多的飞行员中,每一名外籍飞行员都可以与其他若干名英国飞行员很好地配合。如何选择配对飞行的飞行员才能使一次派出最多的飞机。对于给定的外籍飞行员与英国飞行员的配合情况,试设计一个算法找出最佳飞行员配对方案,使皇家空军一次能派出最多的飞机。
对于给定的外籍飞行员与英国飞行员的配合情况,编程找出一个最佳飞行员配对方案,使皇家空军一次能派出最多的飞机。
输入输出格式
输入格式:
第 1 行有 2 个正整数 m 和 n。n 是皇家空军的飞行员总数(n<100);m 是外籍飞行员数(m<=n)。外籍飞行员编号为 1~m;英国飞行员编号为 m+1~n。
接下来每行有 2 个正整数 i 和 j,表示外籍飞行员 i 可以和英国飞行员 j 配合。最后以 2个-1 结束。
输出格式:
第 1 行是最佳飞行员配对方案一次能派出的最多的飞机数 M。接下来 M 行是最佳飞行员配对方案。每行有 2个正整数 i 和 j,表示在最佳飞行员配对方案中,飞行员 i 和飞行员 j 配对。如果所求的最佳飞行员配对方案不存在,则输出‘No Solution!’。
输入样例#1:
5 10
1 7
1 8
2 6
2 9
2 10
3 7
3 8
4 7
4 8
5 10
-1 -1
输出样例#1:
4
1 7
2 9
3 8
5 10

分析:
二分图匹配,源点连向1~n权值为1,1~n连向n+1~m权值为1,最后流向汇点权值为1,问题转化为最大流是多少匹配对就是多少。找匹配对的时候,dfs找第一层和第二层之间边权为正的即为匹配对(答案不唯一)。

#pragma GCC optimize(2)
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
typedef long long LL;
using namespace std;

const int MAXN = 200 + 5;
const int INF = 0x3f3f3f3f;
int head[MAXN], dist[MAXN], vis[MAXN];
int cur[MAXN], ma[MAXN];
int top = 0;

struct Edge {
    int to, cap, flow, next;
}edge[MAXN * 10];

void init() {
    top = 0;
    memset(head, -1, sizeof(head));
    memset(vis, 0, sizeof(vis));
    memset(ma, 0, sizeof(ma));
}

void addedge(int a, int b, int c) {
    Edge E1 = {b, c, 0, head[a]};
    edge[top] = E1;
    head[a] = top++;
    Edge E2 = {a, 0, 0, head[b]};
    edge[top] = E2;
    head[b] = top++;
}

bool BFS(int st, int ed) {
    memset(dist, -1, sizeof(dist));
    memset(vis, 0, sizeof(vis));
    queue<int> que;
    que.push(st);
    vis[st] = 1;
    dist[st] = 0;
    while(!que.empty()) {
        int u = que.front();
        que.pop();
        for(int i = head[u]; i != -1; i = edge[i].next) {
            Edge E = edge[i];
            if(!vis[E.to] && E.cap > E.flow) {
                dist[E.to] = dist[u] + 1;
                vis[E.to] = 1;
                if(E.to == ed) return true;
                que.push(E.to);
            }
        }
    }
    return false;
}

int DFS(int x, int a, int ed) {
    if(x == ed || a == 0) return a;
    int flow = 0, f;
    for(int& i = cur[x]; i != -1; i = edge[i].next) {
        Edge& E = edge[i];
        if(dist[E.to] == dist[x] + 1 && (f = DFS(E.to, min(a, E.cap - E.flow), ed)) > 0) {
            E.flow += f;
            edge[i^1].flow -= f;
            flow += f;
            a -= f;
            if(a == 0) break;
        }
    }
    return flow;
}

int Maxflow(int st, int ed) {
    int flow = 0;
    while(BFS(st, ed)) {
        memcpy(cur, head, sizeof(head));
        flow += DFS(st, INF, ed);
    }
    return flow;
}

int n, m, a, b;
int dfs2(int x) { //寻找匹配对,判断边是否有流量
    for(int j = x; j <= n; ++j) {
        for(int i = head[j]; ~i; i = edge[i].next) {
            if(edge[i].to > n && edge[i].to <= m && edge[i].flow > 0) {
                ma[j] = edge[i].to;
            }
        }
    }
}

int main()
{
    init();
    scanf("%d %d", &n, &m);
    for(int i = 1; i <= m; ++i) {
        if(i <= n) addedge(0, i, 1);
        else addedge(i, m + 1, 1);
    }
    while(scanf("%d %d", &a, &b) && (a + b) != -2) {
        addedge(a, b, 1);
    }
    int ans = Maxflow(0, m + 1);
    memset(vis, 0, sizeof(vis));
    if(!ans) printf("No Solution!\n");
    else {
        printf("%d\n", ans);
        dfs2(1);
        for(int i = 1; i <= n; ++i) {
            if(!ma[i]) continue;
            printf("%d %d\n", i, ma[i]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_36368339/article/details/80584757