首页 > 试题广场 >

分割等和子集

[编程题]分割等和子集
  • 热度指数:2076 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 256M,其他语言512M
  • 算法知识视频讲解
给定一个只包含正整数的数组 nums ,请问能否把这个数组取出若干个数使得取出的数之和和剩下的数之和相同。

数据范围: , 数组中的元素满足
示例1

输入

[1,5,11,5]

输出

true

说明

分割为 [1,5,5] 和 [11]   
示例2

输入

[1,2,3,5]

输出

false
最基础的01背包问题
import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return bool布尔型
     */
    public boolean partition (int[] nums) {
        // write code here
        int sum = 0;
        for(int i = 0; i < nums.length; i++){
            sum += nums[i];
        }
        if(sum % 2 != 0){
            return false;
        }
        int n = sum / 2;
        int[][] dp = new int[nums.length][n + 1];
        for(int i = 1; i < nums.length; i++){
            for(int j = 1; j <= n; j++){
                // 可以放
                if(j >= nums[i]){
                    dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - nums[i]] + nums[i]);
                }else{
                     dp[i][j] = dp[i - 1][j];
                }
            }
        }
        return dp[nums.length - 1][n] == n;
    }
}


发表于 2022-09-14 17:52:15 回复(0)

暴力递归

首先从左往右进行尝试,写出暴力递归版本。
  1. rest<0时,表示本次凑数方案失败;
  2. rest=0时,刚好凑齐,本次凑数方案是成功的;
  3. 否则看本轮取哪个数nums[i],下一轮需要凑rest-nums[i]
private boolean recurrent(int[] nums, int depth, int rest) {
    if(rest < 0) return false;
    if(rest == 0) return true;    // base case
    for(int i = depth; i < nums.length; i++){
        if(recurrent(nums, i, rest - nums[i])) return true;
    }
    return false;
}

动态规划 

递归函数有depthrest两个可变参数,因此本质上是个二维动态规划问题。根据递归的逻辑,就可以改成动态规划版本,递归中的函数调用就是动态规划中从dp数组中取值。
我们可以看,在计算dp[i][rest]时,需要依赖左边的值,因此动态规划填表的顺序:从上往下(从下往上也可以,这个顺序无所谓的情况下还可以进行空间压缩的优化)从左往右。
import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return bool布尔型
     */
    public boolean partition (int[] nums) {
        // write code here
        int sum = 0;
        for(int i = 0; i < nums.length; i++){
            sum += nums[i];
        }
        // 无法均分
        if(sum % 2 != 0){
            return false;
        }
        int target = sum / 2;
        boolean[][] dp = new boolean[nums.length][target + 1];
        for(int i = 0; i < nums.length; i++){
            dp[i][0] = true;    // base case
        }
        for(int i = 0; i < nums.length; i++){
            for(int rest = nums[i]; rest <= target; rest++){
                dp[i][rest] |= dp[i][rest - nums[i]];
            }
        }
        return dp[0][target];
    }
}

编辑于 2021-12-13 21:34:49 回复(0)

问题信息

难度:
2条回答 1893浏览

热门推荐

通过挑战的用户

查看代码