题意:有n头牛, 安排到m个棚里住。每头牛对每个棚都有一个好感度排名。主人为了使得这些牛尽可能满意,求获得最低好感度的牛和获得最高好感度的牛的最小好感度差值(即好感度跨度最小)。
题解:二分+二分图多重匹配
每头牛只能进一个棚,每个棚能容纳多头牛,二分图多重匹配问题。
接下来二分枚举好感度区间,根据区间确定牛和棚之间的连线,然后跑多重匹配即可。
还有就是算最大多重匹配的时候,由于要让所有牛都有棚,若当前区间无法满足这头牛,直接break,会省好多时间。
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
const int MAXN = 1510;
const int MAXM = 50;
int n, b, li;
int uN, vN;
int g[MAXN][MAXM];
int linker[MAXM][MAXN];
bool used[MAXM];
int num[MAXM];//右边最大的匹配数
bool dfs(int u, int l, int r) {
for (int vv = l; vv <= r; vv++) {
int v = g[u][vv]; //g存矩阵,直接取
if (!used[v]) {
used[v] = true;
if (linker[v][0] < num[v]) {
linker[v][++linker[v][0]] = u;
return true;
}
for (int i = 1; i <= num[v]; i++)
if (dfs(linker[v][i], l, r)) {
linker[v][i] = u;
return true;
}
}
}
return false;
}
int hungary(int mid) {
for (int l = 1; l <= b - mid + 1; l++) {
int r = l + mid - 1;
int res = 0;
for (int i = 1; i <= vN; i++)
linker[i][0] = 0;
for (int u = 1; u <= uN; u++) {
memset(used, false, sizeof(used));
if (!dfs(u, l, r)) break; //由于只要判断,不满足直接break不继续求,这样会省很多时间
else res++;
}
if (res == n) return true;
}
return false;
}
int main() {
scanf("%d%d", &n, &b);
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= b; j++) {
scanf("%d", &g[i][j]);
}
}
for (int i = 1; i <= b; i++) {
scanf("%d", &num[i]);
}
uN = n, vN = b;
int l = 1, r = b, mid, ans;
while (l <= r) {
int mid = (l + r) / 2;
if (hungary(mid)) {
ans = mid;
r = mid - 1;
}
else
l = mid + 1;
}
printf("%d\n", ans);
return 0;
}