分割等和子集
给你一个 只包含正整数 的 非空 数组 nums
。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
题解 动态规划
如果可以等分成两个和相同的子集,总和一定是偶数,可以根据这一性质作为条件对一部分进行判断返回;
建立一个布尔类型的dp
容器,其中dp[i]
中的值表示dp[i]是否存在子集的和为i
,最后返回dp[sum / 2]
里面的值即可;
先对num[0]
进行判断,如果小于等于目标值,则dp[nums[0]] = true;
在循环的时候利用利用dp[j] = dp[j] || dp[j - nums[i]];
进行求值;
细节
一定要逆序循环,因为在一维情况下,是根据 dp[j] || dp[j - nums[i]]
来推d[j]
的值,如不逆序,就无法保证在外循环 i 值保持不变 j 值递增的情况下,dp[j - num[i]]
的值不会被当前所放入的nums[i]
所修改,当j值未到达临界条件前,会一直被nums[i]
影响,也即是可能重复的放入了多次nums[i]
,为了避免前面对后面产生影响,故用逆序。
例如:举个例子,数组为[2,2,3,5]
,要找和为6
的组合,i = 0
时,dp[2]
为真,当i自增到1,j = 4
时,nums[i] = 2,dp[4] = dp[4] || dp[4 - 2]
为true,当i不变,j = 6
时,dp[6] = dp [6] || dp [6 - 2],
而dp[4
]为true,所以dp[6] = true
,显然是错误的。 故必须得纠正在正序情况下,i值不变时多次放入nums[i]的情况。
完整代码
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum = 0;
int len = nums.size();
if( len < 2)
return false;
for(int i = 0; i < len; ++i){
sum += nums[i];
}
if(sum % 2)
return false;
int target = sum / 2;
vector<bool> dp(target + 1, false);
dp[0] = true;
if(nums[0] <= target){
dp[nums[0]] = true;
}
else{
return false;
}
for(int i = 1; i < len; ++i){
for(int j = target; j >= nums[i]; --j)
dp[j] = dp[j] || dp[j - nums[i]];
}
return dp[target];
}
};