416. Partition Equal Subset Sum

一、题目

  1、审题 

  

 

  2、分析

    给出一个整形数组,判断是否能将其拆分成两个元素和相等的子数组。

 

二、解答

  1、思路

    ① 将数组所有元素求和得到 sum, 若 sum %2 == 1 ,则不可切分。

    ② 将 sum 除以 2,即 sum /= 2; 则 问题变为:从 数组中取出若干个元素,使得其和为 sum。即 0/1 背包问题。

    ③ 采用布尔型二维动态数组 dp[][];  dp[i][j]: 表示数组中前 i 个元素中是否能存在某个组合使得元素和为 j 。

        若 dp[i][j],取出的组合中不包含数组第 i 个元素,即 nums[i - 1]。则 dp[i][j] = dp[i - 1][j]

        若 dp[i][j] 取出的组合中包含数组第 i 个元素, 即 nums[i - 1]。则 dp[i][j] = dp[i - 1][ j - nums[i - 1]];

      所以 dp[i][j] = dp[i-1][j]  ||  dp[i-1][j - nums[i - 1]]

    最终返回结果为 nums[n][sum] ; 其中 n 为数组长度。

    public boolean canPartition(int[] nums) {
        int sum = 0;
        for(int n : nums) 
            sum += n;
        if((sum & 1) == 1)
            return false;
        
        sum /= 2;
        int n = nums.length;
        boolean[][] dp = new boolean[n + 1][sum + 1];  // dp[i][j]: 前 i 个数能否有组合可以组成 和为 j 。
        for (int i = 0; i < dp.length; i++) {
            Arrays.fill(dp[i], false);
        }
        // 0 个数构成的和为 0
        dp[0][0] = true;
        // i 个数中选取出 x 个,构成的和为 0
        for (int i = 1; i < n + 1; i++) {
            dp[i][0] = true;
        }
        // 0 个数中选取出 x 个,构成的和为 i (i > 0)
        for (int i = 1; i < sum + 1; i++) {
            dp[0][i] = false;
        }
        
        for (int i = 1; i < n + 1; i++) {
            for (int j = 1; j < sum + 1; j++) {
                // 1、不使用 nums[i - 1], 即第 i 个数
                dp[i][j] = dp[i - 1][j];
          // 2、 如果使用 nums[i - 1], 即第 i 个数。 看是否能组成和为 j 
if(j >= nums[i - 1]) dp[i][j] = (dp[i][j] || dp[i - 1][j - nums[i - 1]]); } } return dp[n][sum]; }

 

  优化:

    使用一维数组 boolean 型! 其中 dp[sum] 表示 能否组成和为 sum 的组合。

    public boolean canPartition2(int[] nums) {
        int sum = 0;
        
        for (int num : nums) {
            sum += num;
        }
        
        if ((sum & 1) == 1) {
            return false;
        }
        sum /= 2;
        
        boolean[] dp = new boolean[sum+1];
        
        Arrays.fill(dp, false);
        dp[0] = true;
        for(int num: nums) {
            for(int i = sum; i > 0; i--) {
                if(i >= num)
                    dp[i] = dp[i] || dp[i - sum];
                else 
                    break;
            }
        }
        return dp[sum];
    }

 

posted @ 2019-05-30 12:03  skillking2  阅读(40)  评论(0编辑  收藏