简单的背包——01背包（5）

简单的背包——01背包（5）

题目来源：洛谷 P1064 小A点菜

原题

共十一个测试点

题解

WA代码

当前第 i 种物品恰好为 j 元钱，所以可以只买它自己。但是由于 f[i][0] = 0 ，所以不会算入这种情况。所以我们要把所有的 f[i][0]更新成 1 ，这样就可以计算上面所述的那种情况。

变为一维数组 f[0] = 1;

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e4 + 5;
int f[maxn];
int n, m;
int a;

int main()
{
    memset(f, 0, sizeof(f));
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i)
    {
        scanf("%d", &a);
        for (int j = m; j >= a; --j)
        {
            // f[i][j] = max(f[i - 1][j], f[i - 1][j - a[i]] + 1);
            f[j] = max(f[j], f[j - a] + 1);
        }
    }
    printf("%d", f[m]);
    return 0;
}

错误分析：

初始情况中 f[0] 应该方案数为 1 。

方案数应该为f[j] = f[j] + f[j - a]而不是 f[j - a] + 1。

测试点通过情况：

#1, #3 ~ #11 WA

AC代码1

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e4 + 5;
// f[x] 表示金钱为 x 时的总方案数
int f[maxn];
int n, m;
int a;

int main()
{
    // 初始化
    memset(f, 0, sizeof(f));
    f[0] = 1;

    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i)
    {
        scanf("%d", &a);
        // 消除后效性
        for (int j = m; j >= a; --j)
        {
            // f[j] += f[j - a];
            f[j] = max(f[j], f[j - a] + f[j]);
        }
    }
    printf("%d", f[m]);
    return 0;
}

难以理解f[0] = 1; ，因此产生代码2

AC代码2

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e4 + 5;
// f[x] 表示金钱为 x 时的总方案数
int f[maxn];
int n, m;
int a;

int main()
{
    // 初始化 (无需初始化 f[0] = 1)
    memset(f, 0, sizeof(f));

    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; ++i)
    {
        scanf("%d", &a);
        for (int j = m; j >= a; --j)
        {
            // 处理了 f[0] = 1的情况
            if (j == a)
                ++f[j];
            else
                f[j] += f[j - a];
        }
    }
    printf("%d", f[m]);
    return 0;
}