参考:背包问题九讲
01背包问题
LeetCode:416. 分割等和子集
该题可以转换成01背包问题,数组和的一半作为背包,需要装满背包,即初始化时dp[0][0]=0,dp[0][1...v]=INT_MIN;
,第一版代码如下,时间复杂度O(len * v),空间复杂度O(len * v)。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int len=nums.size();
int v=accumulate(nums.begin(),nums.end(),0);
if (v%2!=0) return false;
else v=v/2+1;
vector<vector<int>> dp(len,vector<int>(v,INT_MIN));
dp[0][0]=0;
for (int i=1;i<len;i++)
for (int j=0;j<v;j++)
dp[i][j]=max(dp[i-1][j],j-nums[i]>=0?dp[i-1][j-nums[i]]+nums[i]:INT_MIN);
if (dp[len-1][v-1]==v-1) return true;
else return false;
}
};
第二版代码,优化了dp数组空间,只需要v+1的一维数组,数组更新从后往前更新。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int len=nums.size();
int v=accumulate(nums.begin(),nums.end(),0);
if (v%2!=0) return false;
else v=v/2+1;
vector<int> dp(v,INT_MIN);
dp[0]=0;
for (int i=1;i<len;i++)
for (int j=v-1;j>=0;j--)
//从后往前更新数组,使用的j和j-nums[i]就是前一个i的
dp[j]=max(dp[j],j-nums[i]>=0?dp[j-nums[i]]+nums[i]:INT_MIN);
if (dp[v-1]==v-1) return true;
else return false;
}
};