【BZOJ 1042】【HAOI 2008】硬币购物【DP+容斥】

Description

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

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

题解

巧妙地利用容斥
ans = 得到面值S的超过限制的方案数 – 第1种硬币超过限制的方案数 – 第2种硬币超过限制的方案数 – 第3种硬币超过限制的方案数 – 第4种硬币超过限制的方案数 + 第1,2种硬币同时超过限制的方案数 + 第1,3种硬币同时超过限制的方案数 + …… + 第1,2,3,4种硬币全部同时超过限制的方案数。

设F[S]为在不限制硬币数的情况下总和为s的方案数。
∴$F[S] = sum(F[S-C[K]],S-C[K] >= 0$ &&$K = 1.2.3.4)$

对于第1种硬币超过限制的方案数=
只要要用到D[1]+1枚硬币，剩余的硬币可以任意分配，所以方案数为 F[ S – (D[1]+1)C[1] ]，当且仅当(S – (D[1]+1)C[1])>=0，否则方案数为0。其余情况类似，每次询问只用问16次，所以询问的时间复杂度为O(1)。

代码

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

int t,s;
int c[5],d[5];
long long ans,f[100010];

int main()
{
    for(int i = 1;i <= 4;i++) scanf("%d",&c[i]);
    scanf("%d",&t);
    f[0] = 1;
    for(int i = 1;i <= 4;i++)
        for(int j = c[i];j <= 100000;j++)
            f[j] += f[j - c[i]];
    while(t--)
    {
        for(int i = 1;i <= 4;i++) scanf("%d",&d[i]);
        scanf("%d",&s);
        ans = f[s];
        for(int i = 1;i <= 4;i++)
            if(s - (d[i]+1)*c[i] >= 0) ans -= f[s-(d[i]+1)*c[i]];
        for(int i = 1;i <= 3;i++)
            for(int j = i+1;j <= 4;j++)
                if(s-(d[i]+1)*c[i]-(d[j]+1)*c[j] >= 0) ans += f[s-(d[i]+1)*c[i]-(d[j]+1)*c[j]];
        int x = 0;
        for(int i = 1;i <= 4;i++)
        {
            x = 0;
            for(int j = 1;j <= 4;j++)
                if(i != j) x += (d[j]+1)*c[j];
            if(s >= x) ans -= f[s-x];
        }
        for(int i = 1;i <= 4;i++)
            x += (d[i]+1)*c[i];
        if(s >= x) ans += f[s-x];
        printf("%lld\n",ans);
    }
    return 0;
}