leetcode416 分隔等和子集(Medium)

题目来源:leetcode416 分隔等和子集

题目描述:

给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。

注意:

每个数组中的元素不会超过 100
数组的大小不会超过 200

示例 1:

输入: [1, 5, 11, 5]

输出: true

解释: 数组可以分割成 [1, 5, 5] 和 [11].

示例 2:

输入: [1, 2, 3, 5]

输出: false

解释: 数组不能分割成两个元素和相等的子集.

解题思路:

0-1背包的变形,见代码注释

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum=0,i,j,n=nums.size();
        for(i=0;i<n;i++){
            sum+=nums[i];
        }
        if(sum%2!=0) return false;//奇数则肯定不能分隔
        sum/=2;
        vector<vector<bool>> dp(n+1,vector<bool>(sum+1,false));//dp[i][j]表示前i个物品装入容量为j的背包,是否能恰好装满
        for(i=0;i<=n;i++){
            dp[i][0]=true;//背包容量为0时,默认为true
        }
        for(i=1;i<=n;i++){
            for(j=1;j<=sum;j++){
                if(j<nums[i-1]){
                    dp[i][j]=dp[i-1][j];//第i个物品下标是i-1,由于物品重量比背包容量大,不能装入,则等于dp[i-1][j]
                }
                else{
                    dp[i][j]=dp[i-1][j]||dp[i-1][j-nums[i-1]];//可以选择装入第i个物品,也可以不装入
                }
            }
        }
        return dp[n][sum];
    }
};

posted @ 2020-08-04 16:22  拉里拉里啦啦  阅读(206)  评论(0编辑  收藏  举报