P1450 [HAOI2008] 容斥原理 + DP

最新推荐文章于 2022-08-18 00:21:01 发布

SHOHOKUKU

最新推荐文章于 2022-08-18 00:21:01 发布

阅读量983

点赞数

分类专栏：数学 DP 文章标签：动态规划硬币购物问题方程求解无约束问题 DP算法

本文链接：https://blog.youkuaiyun.com/neweryyy/article/details/123882236

版权

数学同时被 2 个专栏收录

164 篇文章

订阅专栏

159 篇文章

订阅专栏

题意

传送门 P1450 [HAOI2008]硬币购物

题解

求满足 $\sum x_k c_k = s$ 的非负整数解个数。令 $S_k$ 代表满足 $x_k \leq d_k$ 的解集。问题即求
$∩Sk \cap S_k = \overline{\cup \overline{S_k}}$ 则转换为存在下界约束的方程，以 $\overline{S_k}$ 为例，下界约束 $x_k\geq d_k+1$ ， $s$ 减去下界约束后转换为无约束问题，DP 求解即可。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr int MAXS = 1E5 + 5;
ll dp[5][MAXS];

int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int c[4], n;
    for (int i = 0; i < 4; ++i)
        cin >> c[i];
    dp[0][0] = 1;
    for (int i = 0; i < 4; ++i)
        for (int j = 0; j < MAXS; ++j)
        {
            if (j - c[i] >= 0)
                dp[i + 1][j] += dp[i + 1][j - c[i]];
            dp[i + 1][j] += dp[i][j];
        }
    cin >> n;
    while (n--)
    {
        int d[4], s;
        for (int i = 0; i < 4; ++i)
            cin >> d[i];
        cin >> s;
        ll sum = 0;
        for (int i = 0; i < 1 << 4; ++i)
        {
            ll rem = s, cnt = 0;
            for (int j = 0; j < 4; ++j)
                if (i >> j & 1)
                    ++cnt, rem -= (ll)c[j] * (d[j] + 1);
            if (rem >= 0)
                sum += (cnt & 1 ? -1 : 1) * dp[4][rem];
        }
        cout << sum << '\n';
    }
    return 0;
}