POJ 2112 (二分图多重匹配)

http://poj.org/problem?id=2112

　　题意：给出一些挤奶机、奶牛，以及他们之间距离、权值，每只奶牛都要走到任意一台机器中，每台机器最多为M只奶牛服务，问所有奶牛都完成任务，所走的路程最远的那只奶牛，所走的路程最短可以是多少。（这里注意给出的不一定是最短路径，0代表的是两点之间没有直接路径。）

　　首先求出任意点间的最短路，由于数据量不大，可以用Floyd。注意到，如果将奶牛、机器作为左右两幅图，做一次二分匹配，如果能将所有奶牛匹配完毕，即是完成了任务。但题目要求最远路程最小，需要二分长度，每次二分以是否能完成二分匹配为标准改变上下限。

　　这里的匹配是多重匹配，有两种做法，其一是将每台机器都拆成完全相同的M台机器（拆点），然后进行二分匹配。另一种是直接修改匈牙利算法，完成真正意义上的多重二分匹配。第二种做法可以加深对匈牙利算法的理解，因此我了第二种方法。

代码：

// 297 Ms
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#define N 300
#define inf 1e9
using namespace std;
 
int G1[N][N];        // 原图
vector<int> G2[N];   // 二分图
int match[50][50];   // 匹配结果
int cnt[50];         // 每台机器匹配奶牛的数目
bool mark[50];
 
void Floyd(int n) {
    for (int k = 1; k <= n; k++) {
        for (int i = 1; i <= n; i++)
            for (int j = 1; j <= n; j++)
                if (G1[i][k] + G1[k][j] < G1[i][j])
                    G1[i][j] = G1[i][k] + G1[k][j];
    }
}
 
int DFS(int M, int u) {
    // 改进的匈牙利算法，用于二分多重匹配
    for (int i = 0; i < G2[u].size(); i++) {
        int v = G2[u][i];
        if (mark[v]) continue;
        mark[v] = 1;
        if (cnt[v] < M) {
            match[v][cnt[v]++] = u;
            return 1;
        }
        for (int j = 0; j < M; j++) {
            if (DFS(M, match[v][j])) {
                match[v][j] = u;
                return 1;
            }
        }
    }
    return 0;
}
 
int main() {
    int K, C, M;
    while (scanf("%d %d %d", &K, &C, &M) != EOF) {
        for (int i = 1; i <= K+C; i++) {
            for (int j = 1; j <= K+C; j++) {
                scanf("%d", &G1[i][j]);
                if (G1[i][j] == 0) G1[i][j] = inf;
            }
        }
        Floyd(K+C);   // 计算最短路
 
        int ans;
        int low = 0, high = inf;
        while (low <= high) {
	//  二分长度
            int mid = (low + high) >> 1;
            memset(G2, 0, sizeof(G2));
            memset(cnt, 0, sizeof(cnt));
            memset(match, 0, sizeof(match));
            // 构造二分图
            for (int i = K+1; i <= K+C; i++) {
                for (int j = 1; j <= K; j++) {
                    if (G1[i][j] <= mid)
                        G2[i].push_back(j);
                }
            }
 
            // 二分图多重匹配
            int cnt = 0;
            for (int i = K+1; i <= K+C; i++) {
                memset(mark, 0, sizeof(mark));
                cnt += DFS(M, i);
            }
            if (cnt == C) {
                high = mid - 1;
                ans = mid;
            }
            else low = mid + 1;
        }
        printf("%d\n", ans);
    }
    return 0;
}