CF1316E Team Building（贪心 + 状压 dp）

Description

有 $n$ 个人，每个人当观众让俱乐部的实力多 $a_i$。第 $i$ 个人当第 $j$ 个选手会让俱乐部的实力值多 $s_{i,j}$。求选 $k$ 个人当观众，选出 $p$ 个不同选手的最大实力值。

$2\le n\le 10^5,1\le p\le 10^7$。

Solution

因为 $p$ 很小，所以令 $f_{i,j}$ 为前 $i$ 个人选手状态为 $p$ 的最大实力。将选手按当观众的实力从大到小排序。考虑前 $k+p$ 个人，如果还没有选完 $k$ 个观众即为 $i-j$ 中 $1$ 的个数 $\le k$ 而且它不是选手，那么它一定是观众。

对于选手的转移简单，枚举一位 $1$ 变成 $0$ 转移即可。初始值一定是无穷小，再将 $f_{0,0}$ 赋值为 $0$。最后答案为 $f_{n,(1<<p)-1}$。时间复杂度为 $O(n^2\times 2^n)$。

Code

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e5 + 5, INF = 0x3f3f3f3f;
inline int read() {
	int x = 0, f = 0; char ch = 0;
	while (!isdigit(ch)) f |= ch == '-', ch = getchar();
	while (isdigit(ch)) x = (x << 3) + (x << 1) + (ch ^ 48), ch = getchar();
	return f ? -x : x;
}
struct node {
	int gz, xs[8];
}a[N];
int n, p, k;
bool cmp(node a, node b) {
	return a.gz > b.gz;
}
int f[N][(1 << 7) + 1];
signed main() {
	n = read(), p = read(), k = read();
	for (int i = 1; i <= n; i++) a[i].gz = read();
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= p; j++)
			a[i].xs[j] = read();
	sort(a + 1, a + n + 1, cmp);
	memset(f, 0x80, sizeof(f));
	f[0][0] = 0;
	for (int i = 1; i <= n; i++)
		for (int j = 0; j < (1 << p); j++) {
			if (i <= k + p && i - __builtin_popcount(j) <= k) f[i][j] = f[i - 1][j] + a[i].gz;
			else f[i][j] = f[i - 1][j];
			for (int k = 1; k <= p; k++)
				if (j & (1 << (k - 1))) f[i][j] = max(f[i][j], f[i - 1][j & (~(1 << (k - 1)))] + a[i].xs[k]);
		}
	printf("%lld\n", f[n][(1 << p) - 1]);
	return 0;
}