(2017多校训练第五场)HDU - 6092 Rikka with Subset 0-1背包

题目链接：点击打开链接

题意不再赘述。

下面来讲解一下第一个样例是怎么推导出答案的。

2 3

1 1 1 1

第一个数字和最后一个数字肯定是1，不需要关注它们。

下面来看第二个数字（去掉首尾之后的第一个非零数字），也就是子集和为1的数目只有一个，很容易就知道答案中有一个1。

然后再看第二个非零数字，也就是子集和为2的数目只有一个，我们已经知道了答案中有一个1，如果能够求出已知的答案能够组成多少个2（假设是x个），那么我们就知道答案中有1 - x个2了，在这里1 - x = 1，也就是答案中有一个2。

所以最后结果就是1 2。

我们的方法就是：假设已知答案中有x个1，y个2，z个3，当前非零数字为5，其数目为num个。利用背包的方法求出前三种数字能够组成cnt个5，那么答案中5的数目就是num - cnt个。

然后更新dp数组，继续寻找下一个非零数字。

注意：千万不要写成多重背包，如果是多重背包，3个1组成2的方案只有一种，但是实际上是有三种的。也就是多重背包没考虑顺序。

代码如下：

#include <bits/stdc++.h>

using namespace std;
typedef long long int LL;
LL dp[10005];
LL b[10005];
int ans[105], cnt;

int main()
{
    //freopen("test.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    cin.sync_with_stdio(false);
    int T;
    cin >> T;
    while (T--)
    {
        int n, m;
        cin >> n >> m;
        for (int i = 0; i <= m; i++)
            cin >> b[i];
        memset(dp, 0, sizeof(dp));
        int cnt = 0;
        dp[0] = 1;
        for (int i = 1; i <= m; i++)
        {
            if (b[i])
            {
                int num = b[i] - dp[i];
                for (int j = 0; j < num; j++)
                {
                    ans[cnt++] = i;
                    for (int v = m; v >= i; v--)
                        dp[v] += dp[v - i];
                }
            }
        }
        for (LL i = 0; i < cnt; i++)
            cout << ans[i] << ((i < cnt - 1) ? ' ' : '\n');
    }
    return 0;
}