CodeForces - 1316E Team Building

Solution

博主先开始想的是 $f[i][j]$ 表示选到第 $i$ 个人，位置选择情况为 $j$ 的最大力量（$i$ 如果选了位置也包含在里面）。然后把人按 $a$ 值从大到小排个序，然后选到某个人进行比赛，如果那个人在当前能选的前 $k$ 大之列，就抛弃他在前 $k$ 大的地位，在后面找个能选的 $k+1$ 大塞进去。但是这是错的，因为有可能这次最优选择到后面不会推出最优解。

正解的预处理和我的一样 （其实就是误打误撞）。只是枚举 $i$ 时是按照 $a$ 值从大到小枚举。这样的顺序就有了一个性质：我们可以通过枚举 $i,j$ 知道当前观众的个数（这其实不太准确，应该是观众与啥也不干的人数）。当 $i-1-j中1的个数\le k-1$，说明 $i$ 是可以当观众的（当观众肯定是前面不比赛的人先当）。

然后就瞎搞搞。

Code

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;

const int N = 1e5 + 5;

int n, p, k, a[N], s[N][10], pos[N];
ll f[N][1 << 7];

int read() {
	int x = 0, f = 1; char s;
	while((s = getchar()) > '9' || s < '0') if(s == '-') f = -1;
	while(s >= '0' && s <= '9') x = (x << 1) + (x << 3) + (s ^ 48), s = getchar();
	return x * f;
}

bool cmp(const int x, const int y) {return a[x] > a[y];}

int main() {
	n = read(), p = read(), k = read();
	for(int i = 1; i <= n; ++ i) a[i] = read(), pos[i] = i;
	sort(pos + 1, pos + n + 1, cmp);
	for(int i = 1; i <= n; ++ i)
		for(int j = 1; j <= p; ++ j)
			s[i][j] = read();
	for(int i = 1; i <= n; ++ i) {
		for(int j = 0; j < (1 << p); ++ j) {
			int aud = i - __builtin_popcount(j) - 1;
			if(aud >= 0 && aud < k) f[i][j] = max(f[i][j], f[i - 1][j] + a[pos[i]]);
			else if(aud >= 0) f[i][j] = max(f[i][j], f[i - 1][j]);
			if(aud + 1 < 0) break;
			for(int u = 1; u <= p; ++ u)
				if(j & (1 << u - 1)) f[i][j] = max(f[i][j], f[i - 1][j ^ (1 << u - 1)] + s[pos[i]][u]);
		}
	}
	printf("%lld\n", f[n][(1 << p) - 1]);
	return 0;
}