LeetCode 416. 分割等和子集
阿新 • • 發佈:2018-11-11
題目
給定一個只包含正整數的非空陣列。是否可以將這個陣列分割成兩個子集,使得兩個子集的元素和相等。
注意:
每個陣列中的元素不會超過 100
陣列的大小不會超過 200
示例 1:
輸入: [1, 5, 11, 5]
輸出: true
解釋: 陣列可以分割成 [1, 5, 5] 和 [11].
示例 2:
輸入: [1, 2, 3, 5]
輸出: false
解釋: 陣列不能分割成兩個元素和相等的子集.
思路
首先先遍歷一次陣列,得到整個陣列元素的和。若和為奇數,則必然不能分割,若為偶數,則有可能可以。
試過暴力遞迴,還有排序之後的暴力遞迴,都有特殊的樣例使得不能通過,觀察到題目資料的特點,就是全為正,並且和較小,最大也就20000,因此可以考慮用一個bool陣列,記錄下整數陣列中各個子集元素和的可能。
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum = 0;
for(int loop = 0 ; loop < nums.size() ; loop ++){
sum += nums[loop];
}
if(sum % 2 != 0){
return 0;
}
vector<bool>ifCanSum(sum / 2 + 1 , false);
ifCanSum[0] = true;
for(int loop = 0 ; loop < nums.size() ; loop++){
for(int loop1 = ifCanSum.size() - 1 ; loop1 >= 0 ; loop1 --){
if( ifCanSum[loop1] && loop1 + nums[loop] < ifCanSum.size() ){
ifCanSum[ loop1 + nums[loop] ] = true ;
}
}
}
return ifCanSum.back() ;
}
};