Description
有 n n n 个人,每个人当观众让俱乐部的实力多 a i a_i ai。第 i i i 个人当第 j j j 个选手会让俱乐部的实力值多 s i , j s_{i,j} si,j。求选 k k k 个人当观众,选出 p p p 个不同选手的最大实力值。
2 ≤ n ≤ 1 0 5 , 1 ≤ p ≤ 1 0 7 2 \leq n \leq 10^5, 1 \leq p \leq 10^7 2≤n≤105,1≤p≤107。
Solution
因为 p p p 很小,所以令 f i , j f_{i,j} fi,j 为前 i i i 个人选手状态为 p p p 的最大实力。将选手按当观众的实力从大到小排序。考虑前 k + p k + p k+p 个人,如果还没有选完 k k k 个观众即为 i − j i - j i−j 中 1 1 1 的个数 ≤ k \leq k ≤k 而且它不是选手,那么它一定是观众。
对于选手的转移简单,枚举一位 1 1 1 变成 0 0 0 转移即可。初始值一定是无穷小,再将 f 0 , 0 f_{0,0} f0,0 赋值为 0 0 0。最后答案为 f n , ( 1 < < p ) − 1 f_{n,(1 <\!< p) - 1} fn,(1<<p)−1。时间复杂度为 O ( n 2 × 2 n ) O(n ^ 2 \times 2^n) O(n2×2n)。
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;
}