01背包详解(二维基础+一维优化+练习)

目录

1.  01背包

1.1 题目：

1.2 详解

1.2.1 二维解法

1.2.2 一维解法

2.练习——2024睿抗CAIP本科组RC-u5 工作安排

---

1.  01背包

1.1 题目：

题目链接：2. 01背包问题 - AcWing题库

输入样例：

4 5
1 2
2 4
3 4
4 5

输出样例：

8

1.2 详解

1.2.1 二维解法

dp[i][j]表示前i件物品容量为j的最大价值，它可由i-1转移过来，要么第i件物品装要么不装。

如果装不下的话：dp[i][j]=dp[i-1][j]

如果装的下的话：dp[i][j]=dp[i-1][j-v[i]]+w[i]

代码上我写了比较详细的注释：

//二维01书包思路
//不能装下 dp[i][j]=dp[i-1][j]
//能装下(比较装与不装谁最优) dp[i][j]=max(dp[i-1][j],dp[i-1][j-v[i]]+w[i])
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int dp[N][N];
int v[N],w[N]; //体积、价值 
int main(void){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>v[i]>>w[i];
	for(int i=1;i<=n;i++){ //每件物品有两种可能(选与不选)
		for(int j=1;j<=m;j++){  //容量为j(递推) 
			//1.装不下
			if(v[i]>j)
				dp[i][j]=dp[i-1][j];
			else{  //2.装得下，进行比较 
				dp[i][j]=max(dp[i-1][j],dp[i-1][j-v[i]]+w[i]); 
			} 
		} 	
	}
	cout<<dp[n][m]; 
	return 0;
} 

  如果感觉不是很懂的话，可以看一下b站视频：  动态规划DP0-1背包_哔哩哔哩_bilibili

1.2.2 一维解法

dp[j]表示容量为j可装的最大价值。

通过滚动数组，每次对每件物品选择的时候更新dp,得到该次物品(前i次)的最优dp。

代码实现：

//01背包二维变成一维
//思路：
//滚动数组 
//对每个物品仍然有两种可能：选与不选
//dp[j] 表示容量为i时的最大总价值
//对每一个物品进行选择时候，会对其覆盖
//最后得到的dp[j]就可以表示选择前n个物品容量为j得到的最大总价值
#include<bits/stdc++.h>
using namespace std;
const int N=1010;
int dp[N];
int v[N],w[N];  //体积、容量 
int main(void){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>v[i]>>w[i];
	for(int i=1;i<=n;i++){  //每个物品选还是不选
		//每一次迭代更新dp值
		//如果装不下这次的dp值就和上次的一致，不用变化
		//如果装的下，这次dp的值就需要dp[j]和dp[i-v[i]]+w[i]中的最大值
		//必须从后往前更新，因为后面的值需要用到前面的值，如果先更新前面的值的话后面使用的就是被覆盖掉的
		for(int j=m;j>=v[i];j--){ //能装下更新 
			dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
		} 	
	} 
	cout<<dp[m];
	return 0;
} 

如果感觉不是很懂的话，可以看一下b站视频：这个视频带你初探01背包问题part.2 一维数组解法 滚动数组_哔哩哔哩_bilibili

2.练习——2024睿抗CAIP本科组RC-u5 工作安排

题目链接：RC-u5 工作安排 - 2024 睿抗机器人开发者大赛CAIP-编程技能赛-本科组（省赛） (pintia.cn)

输入样例：

3
5
1 2 50
3 3 100
1 5 1
3 2 5000
4 5 30
5
1 2 50
3 3 20
1 5 1
3 2 5000
4 5 30
5
1 2 50
3 3 100
1 5 1
3 2 5000
5 5 800

输出样例：

101
80
800

具体实现代码：

//思路：一维01背包
//按照结束时间从小到大进行排序
//dp[j]表示结束时间刚好为j获得的最大报酬
#include<bits/stdc++.h>
using namespace std;
const int N=5010;
struct node{
	int t,d,p; //花费的时间、截至时间以及报酬 
};
node job[N];
int dp[N];  
//按照截至时间进行排序 
bool cmp(node x,node y){
	return x.d<y.d; 
}
int main(void){
	int x,n;
	cin>>x;
	while(x--){
		cin>>n; 
		for(int i=1;i<=n;i++)
			cin>>job[i].t>>job[i].d>>job[i].p;	
		sort(job+1,job+n+1,cmp);  //排序
		memset(dp,0,sizeof(dp));  //后面对dp赋值了，对下一组数据计算会有影响，所以必须重初始化 
		for(int i=1;i<=n;i++){  //n个工作选与不选
			//如果第i个工作可以选择，则需要更新(从后往前)
			//什么情况下第i个工作可以被选择？
			//1.当前结束时间<=第i件工作的截至时间(保证第i项工作可以在截至时间之前完成）
			//2.当前结束时间>=第i件工作的时长(保证该项工作完成的时间充足) 
			for(int j=job[i].d;j>=job[i].t;j--){
				dp[j]=max(dp[j],dp[j-job[i].t]+job[i].p);
			} 
		}
		int ans=-1;
		//最后报酬最大不一定在截至时间取得 
		for(int i=1;i<=N;i++){
			ans=max(ans,dp[i]);
		}
		cout<<ans;
		if(x!=0) cout<<endl; 
	}
	return 0;
} 

代码都是自己写的，如果有什么问题的话，欢迎指出(*^_^*)

 都看到这里了，点个赞再走吧！！！