bzoj 1042: [HAOI2008]硬币购物（dp+容斥）

1042: [HAOI2008]硬币购物

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 2555   Solved: 1537
[ Submit][ Status][ Discuss]

Description

　　硬币购物一共有4种硬币。面值分别为c1,c2,c3,c4。某人去商店买东西，去了tot次。每次带di枚ci硬币，买s
i的价值的东西。请问每次有多少种付款方法。

Input

　第一行 c1,c2,c3,c4,tot 下面tot行 d1,d2,d3,d4,s,其中di,s<=100000,tot<=1000

Output

　每次的方法数

Sample Input

1 2 5 10 2

3 2 3 1 10

1000 2 2 2 900

Sample Output

4

27

先假设硬币数量是无限的，dp[i]表示用这4种硬币组成i元的情况数

之后每次查询s，因为数量有限

∴答案要减去

第1种硬币超过限制的方案数+第2种硬币超过限制的方案数+

第3种硬币超过限制的方案数+第4种硬币超过限制的方案数，

因为有重复，

所以还要加上第1,2种硬币同时超过限制的方案数+第1,3种硬币同时超过限制的方案数……+

……+第1,2,3,4种硬币 同时超过限制的方案数，即容斥

其中当第一种硬币超过限制时，相当于用了第a[1]+1枚该硬币，其他硬币随便用，

所以方案数位dp[i-(a[1]+1)*c[1]]，其它同理

#include<stdio.h>
#include<string.h>
#define LL long long
LL ans, c[5], a[5], dp[100005];
void Sech(LL x, LL k, LL s)
{
	if(s<0)
		return;
	if(x==5)
	{
		if(k%2==1)  ans -= dp[s];
		else  ans += dp[s];
		return;
	}
	Sech(x+1, k+1, s-(a[x]+1)*c[x]);
	Sech(x+1, k, s);
}
int main(void)
{
	LL i, j, s, n;
	while(scanf("%lld%lld%lld%lld%lld", &c[1], &c[2], &c[3], &c[4], &n)!=EOF)
	{
		memset(dp, 0, sizeof(dp));
		dp[0] = 1;
		for(i=1;i<=4;i++)
		{
			for(j=c[i];j<=100000;j++)
				dp[j] += dp[j-c[i]];
		}
		while(n--)
		{
			for(i=1;i<=4;i++)
				scanf("%lld", &a[i]);
			scanf("%lld", &s);
			ans = 0;
			Sech(1, 0, s);
			printf("%lld\n", ans);
		}
	}
	return 0;
}