Solution
博主先开始想的是 f [ i ] [ j ] f[i][j] f[i][j] 表示选到第 i i i 个人,位置选择情况为 j j j 的最大力量( i i i 如果选了位置也包含在里面)。然后把人按 a a a 值从大到小排个序,然后选到某个人进行比赛,如果那个人在当前能选的前 k k k 大之列,就抛弃他在前 k k k 大的地位,在后面找个能选的 k + 1 k+1 k+1 大塞进去。但是这是错的,因为有可能这次最优选择到后面不会推出最优解。
正解的预处理和我的一样 (其实就是误打误撞)。只是枚举
i
i
i 时是按照
a
a
a 值从大到小枚举。这样的顺序就有了一个性质:我们可以通过枚举
i
,
j
i,j
i,j 知道当前观众的个数(这其实不太准确,应该是观众与啥也不干的人数)。当
i
−
1
−
j
中
1
的
个
数
≤
k
−
1
i-1-j中1的个数\leq k-1
i−1−j中1的个数≤k−1,说明
i
i
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;
}