今天我们来学习 01背包 \text{今天我们来学习\Huge\textcolor{red}{{01背包}}} 今天我们来学习01背包
先来看一道题目
P1048 [NOIP2005 普及组] 采药
题目描述
辰辰是个天资聪颖的孩子,他的梦想是成为世界上最伟大的医师。为此,他想拜附近最有威望的医师为师。医师为了判断他的资质,给他出了一个难题。医师把他带到一个到处都是草药的山洞里对他说:“孩子,这个山洞里有一些不同的草药,采每一株都需要一些时间,每一株也有它自身的价值。我会给你一段时间,在这段时间里,你可以采到一些草药。如果你是一个聪明的孩子,你应该可以让采到的草药的总价值最大。”
如果你是辰辰,你能完成这个任务吗?
输入格式
第一行有 2 2 2 个整数 T T T( 1 ≤ T ≤ 1000 1 \le T \le 1000 1≤T≤1000)和 M M M( 1 ≤ M ≤ 100 1 \le M \le 100 1≤M≤100),用一个空格隔开, T T T 代表总共能够用来采药的时间, M M M 代表山洞里的草药的数目。
接下来的 M M M 行每行包括两个在 1 1 1 到 100 100 100 之间(包括 1 1 1 和 100 100 100)的整数,分别表示采摘某株草药的时间和这株草药的价值。
输出格式
输出在规定的时间内可以采到的草药的最大总价值。
样例 #1
样例输入 #1
70 3
71 100
69 1
1 2
样例输出 #1
3
提示
【数据范围】
- 对于 30 % 30\% 30% 的数据, M ≤ 10 M \le 10 M≤10;
- 对于全部的数据, M ≤ 100 M \le 100 M≤100。
【样例解释】
辰辰可以采第 2 2 2 和 3 3 3 株药材
简化一下题意
你可以在 N N N 时间内才 M M M 株药材
再引到背包上来
背包容量为 N N N ,有 M M M 件物品
第 i i i 件物品的重量为 v i v_i vi ,价值为 w i w_i wi
推导式为 f i , j = m a x ( f i − 1 , j , f i − 1 , j − v i + w i ) f_{i,j} = max(f_{i-1,j},f_{i-1,j-v_i}+w_i) fi,j=max(fi−1,j,fi−1,j−vi+wi)
那我们就可以愉快地写代码啦
先来看 2 2 2 维的
#include <bits/stdc++.h>
using namespace std;
int v[105];
int w[105];
//第i件物品的重量为v[i],价值为w[i]
int f[105][1005];
//f[i][j]表示有i件物品,背包容量为j时的最大价值
int main()
{
int m,n;
//背包容量为m,有n件物品
cin>>m>>n;
for(int i=1;i<=n;i++) cin>>v[i]>>w[i];
//01背包
for(int i=1;i<=n;i++)
{
//遍历n件物品的重量以及价值
for(int j=0;j<=m;j++)
{
//遍历背包的容量
f[i][j] = f[i-1][j];
//不选这件物品
if(j>=v[i]) f[i][j] = max(f[i-1][j],f[i-1][j-v[i]]+w[i]);
//选这件物品,容量必须>= 重量
}
}
cout<<f[n][m];
return 0;
}
如果 n n n 和 m m m 特别大怎么办
那就需要用到 1 1 1 维数组
可以发现 f i , j f_{i,j} fi,j 总是要借助 f i − 1 , j f_{i-1,j} fi−1,j 的力量
如果 f i , j f_{i,j} fi,j 每次都借助自己上一状态下的力量同样可以实现
推导式为 f j = m a x ( f j , f j − v i + w i ) f_j = max(f_j,f_{j-v_i}+w_i) fj=max(fj,fj−vi+wi)
#include <bits/stdc++.h>
using namespace std;
int v[105];
int w[105];
//第i件物品的重量为v[i],价值为w[i]
int f[1005];
//f[i]表示,背包容量为i时的最大价值
int main()
{
int m,n;
//背包容量为m,有n件物品
cin>>m>>n;
for(int i=1;i<=n;i++) cin>>v[i]>>w[i];
//01背包
for(int i=1;i<=n;i++)
{
//遍历n件物品的重量以及价值
for(int j=m;j>=v[i];j--)
{
//遍历背包的容量
//由于如果想要选这件物品,容量必须>= 重量,所以可以把遍历区间缩小到v[i]~m
//不选这件物品不需要操作
f[j] = max(f[j],f[j-v[i]]+w[i]);
//要在查找f[j-v[i]]的时候它还是上一轮的值,如果正序,会出现f[j-v[i]]已经更新的情况,所以要倒序
}
}
cout<<f[m];
return 0;
}