硬币购物 dp + 容斥原理

dp + 容斥原理

• P1450 [HAOI2008] 硬币购物

题意：给4种不同硬币，每次给出这四种硬币的数量与要凑的数s，问有多少种凑法

思路：如果不考虑四种硬币限制，那么直接对四种硬币完全背包即可，然后输出$dp[s]$即可

那么如果考虑那么我们就要减去一种硬币超过的情况再加上两种硬币超过的情况…，那么怎么考虑超过的情况呢

如果给了$d[i]$个硬币，那么我们取$d[i]+1$就可以，那么对应的计数为$dp[s-(d[i]+1)\ast c[i]]$，用容斥原理枚举$2^4$即可

很酷

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int S = 1e5 + 10;
int c[5],d[5],s;
ll dp[S];
int main()
{
	dp[0] = 1;
	for(int i = 1;i <= 4;i++){
		cin >> c[i];
		for(int j = c[i];j <= S - 10;j++){
			dp[j] += dp[j - c[i]];
		}
	}
	int n;cin >> n;
	while(n--){
		for(int i = 1;i <= 4;i++)
			cin >> d[i];
		cin >> s;
		ll ans = 0;
		for(int i = 0;i < (1 << 4);i++){
			int u = s,f = 1;
			for(int j = 1;j <= 4;j++){
				
				if(i & (1 << (j - 1))){
					f = -f;
					u -= (d[j] + 1) * c[j];
				}
			}
			if(u >= 0)
				ans += f*dp[u];
		}
		cout << ans << endl;
	}
}