G31 容斥原理 集合的交

视频链接：https://www.bilibili.com/video/BV1ZK41127tu/

Luogu P1450 [HAOI2008] 硬币购物

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long LL;
int c[4],d[4],n,s;
LL f[100005];

void pack_pre(){ //完全背包预处理
    f[0] = 1;
    for(int i=0; i<4; i++)
      for(int j=c[i]; j<100005; j++)
        f[j] += f[j-c[i]];
}
LL calc(LL s){ //容斥原理
  LL res = 0;
  for(int i=1; i<1<<4; i++){//枚举状态
    LL t = 0, sign = -1;
    for(int j=0; j<4; j++) //过滤状态
      if(i & 1<<j){
        t += c[j]*(d[j]+1);
        sign = -sign;      
      }
    if(s>=t) res += f[s-t]*sign;
  }
  return f[s]-res;
}
int main(){
    for(int i=0; i<4; i++) scanf("%d",&c[i]);
    pack_pre(); 
    scanf("%d", &n);
    while(n--){
        for(int i=0; i<4; i++) scanf("%d",&d[i]);
        scanf("%d",&s);
        printf("%lld\n", calc(s));
    }
    return 0;
}