KEDA#P0456. 01背包问题

题目描述

一个旅行者有一个最多能装 MM 公斤的背包，现在有 nn 件物品，它们的重量分别是$W_1$ ,$W_2$，…,$W_n$,它们的价值分别为 $C_1$,$C_2$,…,$C_n$，求旅行者能获得最大总价值。

输入

第一行：两个整数，M (背包容量，M≤200M≤200 ) 和 N (物品数量，N≤30N≤30 )；

第 2 ~ N+1行：每行二个整数 $W_i$，$C_i$,表示每个物品的重量和价值。

输出

仅一行，一个数，表示最大总价值。

样例

输入数据 1

10 4
2 1
3 3
4 5
7 9

输出数据 1

12

思路

这是动态规划的基础题——01背包，只需套上递推公式“$dp[i-1][j],dp[i-1][j-w[i]]+c[i]$”即可。

code1.0

#include<bits/stdc++.h>
using namespace std;
int m,n,dp[299][985],w[211],c[985];
int main(){
	cin >> m >> n;
	for(int i=1;i<=n;i++)cin >> w[i] >> c[i];
	for(int i=1;i<=n;i++){
		dp[i][0]=0;
		for(int j=1;j<=m;j++){
			dp[0][j]=0;
			if(w[i]>j){
				dp[i][j]=dp[i-1][j];
			}else{
				dp[i][j]=max(dp[i-1][j],dp[i-1][j-w[i]]+c[i]);
			}
		}
	}
	cout << dp[n][m];
	return 0;
}

但是，可不可以优化呢？可以,我们可以用一维的方式来去优化，递推公式：$dp[j]=max(dp[j],dp[j-w[i]]+c[i])$

code2.0

#include<bits/stdc++.h>
using namespace std;
int m,n,dp[299],w[211],v[985];
int main(){
	cin >> m >> n;
	for(int i=1;i<=n;i++)cin >> w[i] >> v[i];
	for(int i=1;i<=n;i++){
		for(int j=m;j>=w[i];j--){
			dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
		}
	}
	cout << dp[m];
	return 0;
}