GMOJ 4336 - 【WC2016模拟】rabbit

题目地址

描述

有 $n$ 种胡萝卜和 $m$ 种干草，第 $i$ 种胡萝卜有 $a_i$ 个，第 $j$ 种干草有 $b_j$ 堆。

一只兔子要吃一个胡萝卜和一堆干草，要求不能有两个兔子选择的胡萝卜和干草组合相同。问最多能养活多少个兔子。

$1\le n,m\le 2.5\times 10^6$，$0\le a_i\le n$，$0\le b_i\le m$

分析

建立超级源 $S$ 与超级汇 $T$，从 $S$ 连出 $n$ 条容量依次为 $a_i$ 的边，$m$ 条连向 $T$ 容量依次为 $b_i$ 的边。左边的 $n$ 个点与右边的 $m$ 个点两两间存在一条容量为 $1$ 的边。

答案即为所建网络图的最大流。

设左侧割边的下标集合为 $A$，右侧割边的下标集合为 $B$，根据最大流最小割定理，易知需最小化 $\sum _{i\in A}{a}_i+\sum _{i\in B}{b}_i+(n-|A|)\times (m-|B|)$。

不难发现 $a_i,b_i$ 需升序选取，且 $|A|$ 单调递减时，$|B|$ 的决策点单调递增，two-pointers 维护一下即可。

时间复杂度 $O(n+m)$

代码

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 2500000;

long long ans, now;
int a[MAXN], b[MAXN];
int n, m, nd, md, noww;

int main() {
	scanf("%d %d %d %d %d %d", &n, &m, &a[1], &nd, &b[1], &md), now = a[1];
	for (int i = 2; i <= n; i++) a[i] = (a[i - 1] * 58 + nd) % (m + 1), now += a[i];
	for (int i = 2; i <= m; i++) b[i] = (b[i - 1] * 58 + md) % (n + 1);
	sort(a + 1, a + n + 1), sort(b + 1, b + m + 1);
	
	ans = (long long)n * m, noww = 1;
	for (int i = n; i >= 0; i--) {
		while (noww <= m && n - i >= b[noww]) now += b[noww] - n + i, noww++;
		ans = min(ans, now), now += m - noww + 1 - a[i];
	}
	
	printf("%lld", ans);
}