【bzoj1042】 HAOI2008硬币购物 容斥原理+背包dp

暴力就是裸的多重背包，但是肯定过不去，我们考虑转化，问题等价于求

c1x1+c2x2+c3x3+c4x4=s （x1<=d1 x2<=d2 x3<=d3 x4<=d4）的正整数解个数
那么设集合S1={x1<=d1} S2={x2<=d2} S3={x3<=d3} S4={x4<=d4} （反正差不多这个意思，里面的内容省略了）
ans=S1∩S2∩S3∩S4
=f[s]-{x1>d1}∪{x2>d2}∪{x3>d3}∪{x4>d4}
=f[s]-f[s-c1(d1+1)]-f[s-c2(d2+1)]-f[s-c3(d3+1)]-f[s-c4(d4+1)]+f[s-c1(d1+1)-c2(d2+1)]……

然后f数组就是完全背包求得的因为取的物品相当于无上限。

看了论文发现好神奇，一开始忘开longlong了。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>

using namespace std;

long long f[100010];
int c[5],d[5];
int m=100000,T,s;
long long ans;

void dfs(int i,int s,int dep)
{
	if (s<0) return;
	if (i==5) {if (dep&1) ans-=f[s]; else ans+=f[s];return;}
	dfs(i+1,s,dep);
	dfs(i+1,s-c[i]*(d[i]+1),dep+1);
}

int main()
{
	for (int i=1;i<=4;i++) scanf("%d",&c[i]);
	f[0]=1;
	for (int i=1;i<=4;i++)
	  for (int j=c[i];j<=m;j++)
	    f[j]=f[j]+f[j-c[i]];
	scanf("%d",&T);
	while (T--)
	{
		for (int i=1;i<=4;i++) scanf("%d",&d[i]);
		scanf("%d",&s);
		ans=0;
		dfs(1,s,0);
		printf("%lld\n",ans);
	}
	return 0;
}