HDU 2426 Interesting Housing Problem（KM完美匹配）
來源：程序員人生 發布時間：2014-11-11 08:06:05 閱讀次數：1986次
HDU 2426 Interesting Housing Problem
題目鏈接
題意：n個學生，m個房間，給定1些學生想住房的喜歡度，找1個最優方案使得每一個學生分配1個房間，并且使得喜歡度最大，注意這題有個坑，就是學生不會住喜歡度為負的房間
思路：很明顯的KM最大匹配問題，喜歡度為負直接不連邊便可
代碼：
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int MAXNODE = 505;
typedef int Type;
const Type INF = 0x3f3f3f3f;
struct KM {
int n;
Type g[MAXNODE][MAXNODE];
Type Lx[MAXNODE], Ly[MAXNODE], slack[MAXNODE];
int left[MAXNODE], right[MAXNODE];
bool S[MAXNODE], T[MAXNODE];
void init(int n) {
this->n = n;
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++)
g[i][j] = -INF;
}
void add_Edge(int u, int v, Type val) {
g[u][v] = max(g[u][v], val);
}
bool dfs(int i) {
S[i] = true;
for (int j = 0; j < n; j++) {
if (T[j]) continue;
Type tmp = Lx[i] + Ly[j] - g[i][j];
if (!tmp) {
T[j] = true;
if (left[j] == ⑴ || dfs(left[j])) {
left[j] = i;
right[i] = j;
return true;
}
} else slack[j] = min(slack[j], tmp);
}
return false;
}
void update() {
Type a = INF;
for (int i = 0; i < n; i++)
if (!T[i]) a = min(a, slack[i]);
for (int i = 0; i < n; i++) {
if (S[i]) Lx[i] -= a;
if (T[i]) Ly[i] += a;
}
}
int km(int tot) {
for (int i = 0; i < n; i++) {
left[i] = ⑴; right[i] = ⑴;
Lx[i] = -INF; Ly[i] = 0;
for (int j = 0; j < n; j++)
Lx[i] = max(Lx[i], g[i][j]);
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) slack[j] = INF;
while (1) {
for (int j = 0; j < n; j++) S[j] = T[j] = false;
if (dfs(i)) break;
else update();
}
}
int ans = 0;
for (int i = 0; i < tot; i++) {
if (right[i] == ⑴) return ⑴;
if (g[i][right[i]] == ⑴) return ⑴;
ans += g[i][right[i]];
}
return ans;
}
} gao;
int n, m, k;
int main() {
int cas = 0;
while (~scanf("%d%d%d", &n, &m, &k)) {
gao.init(max(n, m));
int u, v, w;
while (k--) {
scanf("%d%d%d", &u, &v, &w);
if (w >= 0) gao.add_Edge(u, v, w);
}
printf("Case %d: %d
", ++cas, gao.km(n));
}
return 0;
}
生活不易，碼農辛苦
如果您覺得本網站對您的學習有所幫助,可以手機掃描二維碼進行捐贈
