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;
}