动态规划系列（五） -- 背包问题

动态规划系列（五）-- 背包问题

1、0 - 1 背包问题

题目：给你一个可装载重量为 W 的背包和 N 个物品，每个物品有重量和价值两个属性。其中第 i 个物品的重量为 wt[i]，价值为val[i]，现在让你用这个背包装物品，最多能装的价值是多少？

分析：
第一步：明确状态 和 选择。状态：背包的容量 和可选择的物品； 选择：装进背包 或 不装进背包。
第二步：明确dp数组的定义
第三步：根据选择，思考状态转移逻辑

代码：

#include <iostream>
#include <vector>
using namespace std;

int knapsak(int W, int N, vector<int> &val, vector<int> &wt){
	vector<vector<int> > dp(N+1, vector<int>(W+1, 0));
	for(int i = 1; i <= N; i++){
		for(int j = 1; j <= W; j++){
			if(j < wt[i-1]){
				dp[i][j] = dp[i-1][j];
			}else{
				dp[i][j] = max(dp[i-1][j], dp[i-1][j-wt[i-1]] + val[i-1]);
			}
		}
	}
	return dp[N][W];
}

int main(){
	int W, N;
	cin>>W>>N;
	vector<int> val(N,0);
	vector<int> wt(N,0);
	for(int i = 0; i < N; i++){
		cin>>val[i]>>wt[i];
	}
	int res = knapsak(W, N, val, wt);
	cout<<res;
}

2、子集背包问题

题目：输入一个只包含正整数的非空数组 nums， 判断这个数组是否可以被分割成两个子集，使得两个子集的元素和相等。

分析：转换为背包问题，给一个可装载重量为 sum/2 的背包 和 N 个物品，每个物品的重量为 nums[i]， 现在让你装物品，是否存在一种装法，能够恰好将背包装满？

代码：

#include <iostream>
#include <vector>
using namespace std;

bool canPartition(vector<int> &nums){
	int n = nums.size();
	int sum = 0;
	for(int i = 0; i < n; i++){
		sum += nums[i];
	}
	if(sum % 2 == 1){  //奇数无法平分 
		return false;
	}
	sum = sum/2;
	vector<vector<bool> > dp(n+1, vector<bool>(sum+1, false));
	for(int i = 0; i <= n; i++){
		dp[i][0] = true;
	}
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= sum; j++){
			if(j - nums[i-1] < 0){
				dp[i][j] = dp[i-1][j];
			}else{
				dp[i][j] = dp[i-1][j] || dp[i-1][j-nums[i-1]];
			}
		}
	}
	return dp[n][sum];
}

int main(){
	int n;
	cin>>n;
	vector<int> nums(n,0);
	for(int i = 0; i < n; i++){
		cin>>nums[i];
	}
	bool res = canPartition(nums);
	if(res){
		cout<<"true";
	}else{
		cout<<"false";
	}
	return 0;
}

3、完全背包问题

题目：给定不同面额的硬币 coins 和一个总金额 amount，写一个函数来计算可以凑成总金额的硬币组合数。假设每一种面额的硬币有无限个。

分析：将这个问题转化为背包问题，有一个背包，最大容量是 amount， 有一系列物品 coins，每个物品的重量为 coins[i]， 每个物品的数量是无限的。请问有多少种方法，能够恰好装满？

代码：

#include <iostream>
#include <vector>
using namespace std;

int change(int amount, vector<int> &nums){
	int n = nums.size();
	vector<vector<int> > dp(n+1, vector<int>(amount+1, 0));
	for(int i = 0; i <= n; i++){
		dp[i][0] = 1;
	}
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= amount; j++){
			if(j - nums[i-1] >= 0){
				dp[i][j] = dp[i-1][j] + dp[i][j-nums[i-1]];
			}else{
				dp[i][j] = dp[i-1][j];
			}
		}
	}
	return dp[n][amount];
}

int main(){
	int n, amount;
	cin>>amount>>n;
	vector<int> nums(n,0);
	for(int i = 0; i < n; i++){
		cin>>nums[i];
	}
	int res = change(amount, nums);
	cout<<res;
	return 0;
}