洛谷 P1048 [NOIP2005 普及组] 采药 (01背包)

题目传送门

题目描述

辰辰是个天资聪颖的孩子，他的梦想是成为世界上最伟大的医师。为此，他想拜附近最有威望的医师为师。医师为了判断他的资质，给他出了一个难题。医师把他带到一个到处都是草药的山洞里对他说：“孩子，这个山洞里有一些不同的草药，采每一株都需要一些时间，每一株也有它自身的价值。我会给你一段时间，在这段时间里，你可以采到一些草药。如果你是一个聪明的孩子，你应该可以让采到的草药的总价值最大。”

如果你是辰辰，你能完成这个任务吗？

输入格式

第一行有 $2$ 个整数 $T$（$1\le T\le 1000$）和 $M$（$1\le M\le 100$），用一个空格隔开，$T$ 代表总共能够用来采药的时间，$M$ 代表山洞里的草药的数目。

接下来的 $M$ 行每行包括两个在 $1$ 到 $100$ 之间（包括 $1$ 和 $100$）的整数，分别表示采摘某株草药的时间和这株草药的价值。

输出格式

输出在规定的时间内可以采到的草药的最大总价值。

样例 #1

样例输入 #1

70 3
71 100
69 1
1 2

样例输出 #1

3

提示

【数据范围】

• 对于 $30\%$ 的数据，$M\le 10$；
• 对于全部的数据，$M\le 100$。

【题目来源】

NOIP 2005 普及组第三题

问题分析

这是一道典型的背包题目,题目大意为:

有若干草药,每种草药有它的价值,以及采它需要花费的时间,在给定时间中,采药的最大价值

数据范围( $T\le 1000$ $M\le 100$ )很小,并且每种药只能采一次,可以采用 01背包 解

01背包

01背包通俗点说,是你带来60元进超市,买到尽可能好吃的东西
举个例子(方格内为最大价值)

你的钱数	10	20	30	40	50	60
物品1(10元,1价值)	1	2	3	4	5	6
物品2(20元,3价值)	1	3	4	6	7	9
物品3(30元,7价值)	1	3	7	8	11	14
物品4(40元,15价值)	1	3	7	15	16	19

根据上表,列数组 f[i][j] ,初始状态 f[0][0] 为 0
i 表示当前选到第几个物品
j 表示当前的钱数
f[i][j] 即在j元钱、第i件物品时,买到的最大价值
而因为只有两种状态:买或不买
不买即

f[i][j]=f[i-1][j]

买即

f[i][j]=f[i-1][j-w[i]]+v[i]

解释:
j-w[i]+v[i] 即买了之后在上次在 j-w[i] 元时的最优解加上当前物品本身的钱数

---

AC代码:

/*
 * @Author: snake qzrobot_snake@outlook.com
 * @Date: 2023-03-24 20:32:33
 * @LastEditors: snake
 * @LastEditTime: 2023-04-05 20:13:04
 * @FilePath: \cppDemo\src\main.cpp
 */
#include <iostream>
using namespace std;
int w[105], v[105];
int f[105][1005];
int main() {
    int t, m;
    cin >> t >> m;
    for (int i = 1; i <= m; i++) {
        cin >> w[i] >> v[i];
    }
    for (int i = 1; i <= m; i++)
        for (int j = t; j >= 0; j--) {
            if (j >= w[i]) {
                f[i][j] = max(f[i - 1][j - w[i]] + v[i], f[i - 1][j]);
            } else {
                f[i][j] = f[i - 1][j];
            }
        }
    cout << f[m][t] << endl;
    return 0;
}

优化

f[i][j] = max(f[i - 1][j - w[i]] + v[i], f[i - 1][j]);

观察上面的公式,我们可以发现, f[i-1][1~N] 都是已经处理过的内容了,我们可以用滚动数组阻止一些不必要的空间浪费,每次更新时,将现在的内容辅助到原来的数组上,从二维数组就可以简化成一维数组,复杂度也从
O(N*M)降至O(M),大大增加效率
代码(别直接复制):

/*
 * @Author: snake qzrobot_snake@outlook.com
 * @Date: 2023-03-24 20:32:33
 * @LastEditors: snake
 * @LastEditTime: 2023-04-05 20:23:40
 * @FilePath: \cppDemo\src\main.cpp
 */
#include <iostream>
using namespace std;
int n, c, w[110], v[110], f[10010];
int main() {
    cin >> c >> n;
    for (int i = 1; i <= n; i++) {
        cin >> w[i] >> v[i];
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j >= c; j++) {
            f[j] = max(f[j], f[j - w[i]] + v[i]);
        }
    }
    cout << f[c];
    return 0;
}

如果错了,请不要惊讶,接着往下看:

因为我们在更新f[i]数组时,用的是f[i]和f[i - w[i]]这两个状态,假如我们从1更新到c,那么当我们更新f[i]时,f[i - w[i]]已经被更新过了,它已经不是原来的那个f[i - w[i]]了,很有可能f[i - w[i]]已经买了一包第i种物品,然后我们更新f[i]时调用f[i - w[i]],又买了一包一样的物品，这不符合01背包,一种物品已经用了多次了

怎么办呢?有点人反应很快,倒着买,这样就不会出现重复了
代码:

/*
 * @Author: snake qzrobot_snake@outlook.com
 * @Date: 2023-03-24 20:32:33
 * @LastEditors: snake
 * @LastEditTime: 2023-04-05 18:32:05
 * @FilePath: \cppDemo\src\main.cpp
 */
#include <iostream>
using namespace std;
int n, c, w[110], v[110], f[10010];
int main() {
    cin >> c >> n;
    for (int i = 1; i <= n; i++) {
        cin >> w[i] >> v[i];
    }
    for (int i = 1; i <= n; i++) {
        for (int j = c; j >= 1; j--) {
            f[j] = max(f[j], f[j - w[i]] + v[i]);
        }
    }
    cout << f[c];
    return 0;
}

继续优化:

第二层循环中,j是从 c 枚举到 1 的,可是在 f[1~w[i]] 中,这些内容已经不用处理了,在下一层循环中也用不到,我们可以把第二层循环改改,改成从 c 枚举到 w[i] 即可

最优代码:

/*
 * @Author: snake qzrobot_snake@outlook.com
 * @Date: 2023-03-24 20:32:33
 * @LastEditors: snake
 * @LastEditTime: 2023-04-05 18:32:05
 * @FilePath: \cppDemo\src\main.cpp
 */
#include <iostream>
using namespace std;
int n, c, w[110], v[110], f[10010];
int main() {
    cin >> c >> n;
    for (int i = 1; i <= n; i++) {
        cin >> w[i] >> v[i];
    }
    for (int i = 1; i <= n; i++) {
        for (int j = c; j >= w[i]; j--) {
            f[j] = max(f[j], f[j - w[i]] + v[i]);
        }
    }
    cout << f[c];
    return 0;
}

创作不易,给个点赞关注吧,谢谢