416. 分割等和集(java)

题目描述

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

注意:

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

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

输出: true

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

示例 2:

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

输出: false

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

---

解法一：动态规划

0-1背包：选取物品的容积总量不能超过规定的总量。思路是：物品一个个先，容量也一点一点放大

1. 若nums[i]之前已有元素使得和为j，则dp[i] [j]=true;
2. 若选择nums[i]加入，则它们和为j-nums[i];

状态方程：dp[i] [j]=dp[i-1] [j] or dp[i] [j-nums[i]]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-cF0iSCBF-1594955320348)(D:\leetcode\题目\动态规划\416. 分割等和子集.assets\image-20200717104124807.png)]

code 1

class Solution {
    public boolean canPartition(int[] nums) {
        int n=nums.length;
        int sum=0;
        for(int i:nums){
            sum+=i;
        }
        if(sum%2==1||sum==0){
            return false;
        }
        int value=sum/2;
        boolean[][] ans = new boolean[n][value+1];
        //填充第一行，nums[0]对应的位置
        if(nums[0]<=value){
            ans[0][nums[0]]=true;
        }
        for(int i=1;i<n;i++){
            for(int j=0;j<=value;j++){
                //先填入上一个结果
                ans[i][j]=ans[i-1][j];
                //刚好得到结果
                if(nums[i]==j){
                    ans[i][j]=true;
                    continue;
                }
                //配合之前选择得到结果
                if(nums[i]<j){
                    ans[i][j]=ans[i-1][j]||ans[i-1][j-nums[i]];
                }
            }
        }
        return ans[n-1][value];
    }
}

code 2 剪枝

class Solution {
    public boolean canPartition(int[] nums) {
        int n=nums.length;
        int sum=0;
        for(int i:nums){
            sum+=i;
        }
        if(sum%2==1||sum==0){
            return false;
        }
        int value=sum/2;
        boolean[][] ans = new boolean[n][value+1];
        //填充第一行，nums[0]对应的位置
        if(nums[0]<=value){
            ans[0][nums[0]]=true;
        }
        for(int i=1;i<n;i++){
            for(int j=0;j<=value;j++){
                //先填入上一个结果
                ans[i][j]=ans[i-1][j];
                //配合之前选择得到结果
                if(nums[i]<=j){
                    ans[i][j]=ans[i-1][j]||ans[i-1][j-nums[i]];
                }
                //进行剪枝
                if(ans[i][value]==true){
                    return true;
                }
            }
        }
        return ans[n-1][value];
    }
}

复杂度同上

code 3 压缩一维

• 当前行总是参考了它上面一行 “头顶上” 那个位置和“左上角”某个位置的值。因此，我们可以只开一个一维数组，从后向前依次填表即可。

class Solution {
    public boolean canPartition(int[] nums) {
        int n=nums.length;
        int sum=0;
        for(int i:nums){
            sum+=i;
        }
        if(sum%2==1||sum==0){
            return false;
        }
        int value=sum/2;
        boolean[] ans = new boolean[value+1];
        //填充第一行，nums[0]对应的位置
        if(nums[0]<=value){
            ans[nums[0]]=true;
        }
        for(int i=1;i<n;i++){
            //nums[i]<=j为了配合j-nums[i]>=0
            for(int j=value;nums[i]<=j;j--){
                if(ans[value]){
                    return true;
                }
                ans[j]=ans[j]||ans[j-nums[i]];
            }
        }
        return ans[value];
    }
}