首页 > 代码库 > [BZOJ 1042] [HAOI2008] 硬币购物 【DP + 容斥】

[BZOJ 1042] [HAOI2008] 硬币购物 【DP + 容斥】

题目链接:BZOJ - 1042

 

题目分析

首先 Orz Hzwer ,代码题解都是看的他的 blog。

这道题首先使用DP预处理,先求出,在不考虑每种硬币个数的限制的情况下,每个钱数有多少种拼凑方案。

为了避免重复的方案被转移,所以我们以硬币种类为第一层循环,这样阶段性的增加硬币。

一定要注意这个第一层循环要是硬币种类,并且初始 f[0] = 1。

f[0] = 1;for (int i = 1; i <= 4; ++i) {	for (int j = B[i]; j <= MaxS; ++j) {		f[j] += f[j - B[i]];	}}

之后对于每个询问 (A1, A2, A3, A4, S) ,根据容斥原理,我们要求的答案 Ans 就是 f[S] - (硬币1超限制的方案数) - (硬币2超限制的方案数) - (硬币3超限制的方案数) - (硬币4超限制的方案数) + (硬币1,2超限制的方案数) + (硬币1,3超限制的方案数) + (硬币1,4超限制的方案数) + .... - (硬币1,2,3超限制的方案数) - ... + (硬币1,2,3,4超限制的方案数) 。  

怎样求硬币1超限制的方案数呢?我们只要先固定取 (A1+1) 个硬币1,剩余的钱数随便取就可以了,就是 f[S - (A1+1) * V[1]] 。

其余的情况都类似。

容斥的部分使用搜索实现。

 

代码

#include <iostream>#include <cstdio>#include <cstdlib>#include <cmath>#include <algorithm>#include <cstring>using namespace std;const int MaxN = 7, MaxS = 100000;int n, S;int A[MaxN], B[MaxN];typedef long long LL;LL Ans;LL f[MaxS + 5];void DFS(int x, int k, int Sum) {	if (Sum < 0) return;	if (x == 5) {		if (k & 1) Ans -= f[Sum];		else Ans += f[Sum];		return;	}	DFS(x + 1, k + 1, Sum - (A[x] + 1) * B[x]);	DFS(x + 1, k, Sum);}int main() {	for (int i = 1; i <= 4; ++i) scanf("%d", &B[i]);	scanf("%d", &n);	f[0] = 1;	for (int i = 1; i <= 4; ++i) {		for (int j = B[i]; j <= MaxS; ++j) {			f[j] += f[j - B[i]];		}	}	for (int i = 1; i <= n; ++i) {		for (int j = 1; j <= 4; ++j) scanf("%d", &A[j]);		scanf("%d", &S);		Ans = 0ll;		DFS(1, 0, S);		printf("%lld\n", Ans);	}	return 0;}

  

[BZOJ 1042] [HAOI2008] 硬币购物 【DP + 容斥】