1. 程式人生 > >LeetCode 416. 分割等和子集

LeetCode 416. 分割等和子集

題目

給定一個只包含正整數的非空陣列。是否可以將這個陣列分割成兩個子集,使得兩個子集的元素和相等。

注意:

每個陣列中的元素不會超過 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() ; } };