「CF744C」Hongcow Buys a Deck of Cards「状压 DP」

题意

你有\(n\)个物品，物品和硬币有\(A\)，\(B\)两种类型，假设你有\(M\)个\(A\)物品和\(N\)个\(B\)物品

每一轮你可以选择获得\(A, B\)硬币各\(1\)个，或者（硬币足够）花\(\max(a_i - M, 0)\)个\(A\)，\(\max(b_i - N, 0)\)个\(B\)买\(i\)这个物品

问买到所有物品最少要多少轮

题解

巧妙的\(dp\)，考虑间接设计状态

\(f[S][A] = B\)表示\(S\)这个集合买过了，\(A\)类花\(\sum a_i-A\)个硬币， \(B\)类花\(\sum b_i - B\)个硬币，这里的\(B\)取最大值

考虑可以先凑硬币，所有物品都最后买，统计一下答案

#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;

const int N = 16; 
const int M = N * N + 5;

int n, suma, sumb, t[N], lg[1 << N], a[N], b[N];
int cnta[1 << N], cntb[1 << N], f[1 << N][M];

int main() {
	scanf("%d", &n); char s[10]; 
	for(int i = 0; i < n; i ++) {
		scanf("%s%d%d", s, a + i, b + i);
		if(*s == 'R') t[i] = 0;
		if(*s == 'B') t[i] = 1;
		suma += a[i]; sumb += b[i];
	}
	for(int i = 0; i < n; i ++) {
		lg[1 << i] = i;
	}
	for(int i = 1; i < (1 << n); i ++) {
		cnta[i] = cnta[i & (i - 1)] + (t[lg[i & (-i)]] == 0);
		cntb[i] = cntb[i & (i - 1)] + (t[lg[i & (-i)]] == 1);
	}
	memset(f, -1, sizeof f); f[0][0] = 0;
	for(int S = 0; S < (1 << n); S ++) {
		for(int j = 0; j <= 120; j ++) if(~ f[S][j]) {
			for(int i = 0; i < n; i ++) if(!(S >> i & 1)) {
				int &dp = f[S | (1 << i)][j + min(a[i], cnta[S])];
				dp = max(dp, f[S][j] + min(b[i], cntb[S]));
			}
		}
	}
	int ans = 2e9;
	for(int i = 0; i <= 120; i ++) {
		if(~ f[(1 << n) - 1][i]) {
			ans = min(ans, max(suma - i, sumb - f[(1 << n) - 1][i]));
		}
	}
	printf("%d\n", ans + n);
	return 0;
}