Partition Equal Subset Sum (Leetcode 416)

這道題是一道變形題,可以變型到背包。

建立背包,表示是否可以拿到總和的一半。

注意,這道題不能變形到Presum,因為Presum數組的題目,要求數組是連續的(Subarray Sum)。如果不連續(比如求數組1,3,5,7中能否挑出幾個數,使其和為10),這就是背包問題。

另外,問能不能fill到時,用一維數組,從后往前做,如下:

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        if(nums.empty()) return false;
        int sum = 0;
        for(auto it : nums){
            sum += it;
        }
        
        if(sum % 2 != 0) return false;
        int val = sum / 2, size = nums.size();
        
        vector<int> dp(val+1, 0);
        dp[0] = true;
        
        for(int i=1; i<=nums.size(); i++){
            for(int j=val; j>=nums[i-1]; j--){
                dp[j] = dp[j] || dp[j-nums[i-1]];
            }
        }
        return dp[val];
    }
};

滾動數組解法:

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        if(nums.empty()) return true;
        int sum = accumulate(begin(nums), end(nums), 0, plus<int>());
        if(sum % 2 != 0) return false;

        int size = nums.size();
        int val = sum/2;
        vector<vector<int>> dp(2, vector<int>(val+1, 0));
        for(int i=1; i<=size; i++){
            for(int j=1; j<=val; j++){
                if(j < nums[i-1]) dp[i%2][j] = dp[(i-1)%2][j];
                else{
                    dp[i%2][j] = max(dp[(i-1)%2][j], dp[(i-1)%2][j-nums[i-1]] + nums[i-1]);
                }
            }
        }
        return dp[size%2][val] == sum - dp[size%2][val];
        
    }
};
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容