O - Steady Cow Assignment POJ - 3189
题意:n个牛分配牛舍,然后每个牛舍有最大的空间限制,每个牛有喜欢的牛舍,然后让你编程解决最大满意度和最小满意度差值最小。
分析:因为牛舍人数固定我们可以二分满意度差值然后枚举区间,然后看能否达到这个差值即可。
#include <iostream>
#include <cstdio>
#include <stack>
#include <cmath>
#include <set>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
const int INF = 0x3f3f3f3f;
const int maxn = 1510;
int un, vn, g[maxn][maxn], linker[maxn][maxn];
bool used[maxn];
int num[maxn], mid, n, m, t, b, a[maxn][maxn];
string str;
bool dfs(int u, int l, int r)
{
for(int v = l; v <= r; v++)
{
int x = a[u][v];
if(!used[x])
{
used[x] = 1;
if(linker[x][0] < num[x])
{
linker[x][++linker[x][0]] = u;
return true;
}
for(int i = 1; i <= num[x]; i++)
if(dfs(linker[x][i], l, r)){
linker[x][i] = u;
return true;
}
}
}
return false;
}
bool hungry(int l, int r)
{
int res = 0;
for(int i = 0; i <= n; i++) linker[i][0] = 0;
for(int u = 1; u <= n; u++){
memset(used, 0, sizeof(used));
if(!dfs(u, l, r)) return false;
}
return true;
}
bool check()
{
for(int i = 1; i + mid - 1 <= m; i++)
if(hungry(i, i + mid - 1)) return true;
return false;
}
int main()
{
while(~scanf("%d%d", &n, &m))
{
memset(g, 0, sizeof(g));
for(int i = 1; i <= n; i++)
for(int j = 1; j <= m; j++)
scanf("%d", &a[i][j]);
for(int i = 1; i <= m; i++) scanf("%d", &num[i]);
int l = 0, r = m, ans = 0;
while(l <= r)
{
mid = (l + r) / 2;
if(check()) ans = mid, r = mid - 1;
else l = mid + 1;
}
printf("%d\n", ans);
}
return 0;
}