题面
传送门:https://www.luogu.org/problemnew/show/P1450
Solution
这是一道很有意思的在背包里面做容斥的题目。
首先,我们可以很轻松地想到暴力做背包的做法。
就是对于每一次询问,我们都做一次背包。
复杂度O(tot*s*log(di)) (使用二进制背包优化)
显然会T得起飞。
接下来,我们可以换一种角度来思考这个问题。
首先,我们可以假设没有每个物品的数量的限制,那么这样就会变成一个很简单的完全背包问题。
至于完全背包怎么写,我们在这里就不做过多讨论,如有需要,看看代码就能理解了。
完全背包做完后,我们可以得到一个f[i]表示填满i的背包的方案数的数组。
那么,我们接下来可以用容斥来解决不可行的方案的问题。
假设只有1件物品的使用次数超出了所给的数量,假设这件物品是第x件。
那么可以用 f[s-(d[x]+1)*c[x]] 表示这件物品不可行的方案总数。
因为对于花钱数为 s-(d[x]+1)*c[x] 里面的每一种方法,都可以通过使用购买 d[x]+1件的x物品来超出所给的数量。所以 f[s-(d[x]+1)*c[x]] 可以表示该物品不可行方案总数。
那么答案是四件物品不可行方案总数这和吗?
nope
因为我们会重复减去一些东西。例如:一种方案即超出了第一件物品的使用数,也超出了第二件物品的使用数,我们却重复扣除了这种方案两次。
所以说我们这时候就得使用容斥来解决这个问题。
容斥中有一个很基础的定理(我不会证):
对于有n的限制条件的事件,只要其中符合一个条件就算可行,其可行方案总数为:
(符合其中0个(条件的方案数,后同)-符合其中1个+符合其中2个-符合其中3个+符合其中4个-符合其中5个+符合其中6个....)
那么,我们就可以根据这个定理求出不可行的方案总数。
对于这题来说,代码如下:
for(a[1]=0;a[1]<=1;a[1]++) for(a[2]=0;a[2]<=1;a[2]++) for(a[3]=0;a[3]<=1;a[3]++) for(a[4]=0;a[4]<=1;a[4]++) { int cnt=0,t_s=s; for(int j=1;j<=4;j++) if(a[j]==1) cnt++,t_s-=(d[j]+1)*c[j]; if(cnt%2==0) cnt=1; else cnt=-1; if(t_s>=0) ans=ans+cnt*f[t_s]; }
这样写绝对不推荐,因为这样写很丑。
那么,这题就可以AC啦
Code
//Luogu P1450 [HAOI2008]硬币购物 //Aug,27th,2018 //DP+容斥 #include <iostream> #include <cstdio> using namespace std; long long read() { long long x=0,f=1; char c=getchar(); while(!isdigit(c)){if(c==‘-‘) f=-1;c=getchar();} while(isdigit(c)){x=x*10+c-‘0‘;c=getchar();} return x*f; } const int N=100000+1000; long long f[N]; int n,m; long long c[5],d[5]; int main(int argc, char **argv) { for(int i=1;i<=4;i++) c[i]=read(); m=read(); f[0]=1; for(int j=1;j<=4;j++) for(int i=c[j];i<=100000;i++) f[i]+=f[i-c[j]]; for(int i=1;i<=m;i++) { for(int j=1;j<=4;j++) d[j]=read(); int s=read(),a[5]; long long ans=0; for(a[1]=0;a[1]<=1;a[1]++) for(a[2]=0;a[2]<=1;a[2]++) for(a[3]=0;a[3]<=1;a[3]++) for(a[4]=0;a[4]<=1;a[4]++) { int cnt=0,t_s=s; for(int j=1;j<=4;j++) if(a[j]==1) cnt++,t_s-=(d[j]+1)*c[j]; if(cnt%2==0) cnt=1; else cnt=-1; if(t_s>=0) ans=ans+cnt*f[t_s]; } printf("%lld\n",ans); } return 0; }
正解(c++)
原文地址:https://www.cnblogs.com/GoldenPotato/p/9545340.html