剑指 Offer II 101. 分割等和子集

题目描述

给定一个非空的正整数数组 nums ，请判断能否将这些数字分成元素和相等的两部分。

解答

这道题可以换一种表述：给定一个只包含正整数的非空数组，判断是否可以从数组中选出一些数字，使得这些数字的和等于整个数组的元素和的一半。因此这个问题可以转换成「0−1 背包问题」。这道题与传统的「0−1 背包问题」的区别在于，传统的「0−1 背包问题」要求选取的物品的重量之和不能超过背包的总容量，这道题则要求选取的数字的和恰好等于整个数组的元素和的一半。类似于传统的「0−1 背包问题」，可以使用动态规划求解。

参考链接：https://leetcode.cn/problems/NUPfPr/solution/fen-ge-deng-he-zi-ji-by-leetcode-solutio-re1t/

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        bool dp[201][20001]={0};
        int sum[201]={0};
        dp[0][nums[0]]=1;
        dp[0][0]=1;
        int n=nums.end()-nums.begin();
        sum[0]=nums[0];
        for (int i=1; i<n; i++)
        {
            sum[i]=sum[i-1]+nums[i];
            dp[i][nums[i]]=1;
            for (int j=0; j<=sum[i]; j++)
            {
                if (dp[i-1][j]) dp[i][j]=1;
                else
                if (j>=nums[i] && dp[i-1][j-nums[i]])
                dp[i][j]=1;
            }
        }
        if (sum[n-1]%2) return false;
        else return dp[n-1][sum[n-1]/2];
    }
};