这道题解法很清晰,用DP去做,但是我第一次做是直接套背包解法,转移方程也没有变,是的。我就是这么废;
class Solution {
public:
bool canPartition(vector<int>& nums) {
int x=0;
x=3|1;
cout<<x;
int i,j,sum=0,n=nums.size();
for(i=0;i<nums.size();++i)
sum+=nums[i];
if(sum%2)
return false;
sum/=2;
int dp[n+1][sum+1];
memset(dp,0,sizeof(dp));
for(i=1;i<=n;i++)
{
for(j=1;j<=sum;j++)
{
if(nums[i-1]<=j)
dp[i][j]=max(dp[i-1][j],dp[i-1][j-nums[i-1]]+nums[i-1]);
else
dp[i][j]=dp[i-1][j];
if (dp[i][j]==sum)
return true;
}
}
return false;
}
};
一开始我就是不明白为什么设置dp数组为bool数组,或者说我没有想明白怎么去判定能否刚好装满。是的,迷糊了好一会。背包原始问题是求最大,所以是max,这里是问能不能等于(也就是刚好装满),所以用bool就可以了。
而dp[i][j]=dp[i-1][j]|dp[i-1][j-nums[i]]。这个也能理解了,前半部分说的是如果上一行就能等于,下一行自然能等于,或者上一行的j-nums[i]装满,所以这一行选上nums[i]的话,也能装满,二者只要满足一个即可。
关键在于初始状态的设定,很显然dp[0][]除了dp[0][0]都是false,而dp[][0]都是true.
而关于压缩变量的,因为前面的会影响后面的,如果从前遍历,则需要两行才可以,如果从后面遍历,则需要一行即可。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum=0,n=nums.size();
for(int i=0;i<nums.size();++i)
sum+=nums[i];
if(sum%2||n<2)
return false;
sum/=2;
bool dp[sum+1];
memset(dp,false,sizeof(dp));
dp[0]=true;
for(int i=1;i<=n;i++)
{
for(int j=sum;j>0;--j)
//for(int j=sum;j>=nums[i-1];--j)截止到nums[i-1]可以少走几步
if(nums[i-1]<=j)
dp[j]|=dp[j-nums[i-1]];
/*也可以中间加上下面代码提前中止
if (dp[sum]==true)
return true;*/
}
return dp[sum];
}
};
这里看到还有用位运算去解的,实际上还是动态规划的思想,他把数组拉的更长,实际也没有必要,其实跟用bool一样。