美文网首页动态规划
背包问题(01背包+leetcode416)

背包问题(01背包+leetcode416)

作者: Mereder | 来源:发表于2019-04-28 22:04 被阅读9次

    动态规划合集:

    1.矩阵链乘法
    2.投资组合问题
    3.完全背包问题
    4.01背包问题
    5.最长公共子序列

    例题4——01背包

    题目描述(Leetcode 416)

    Given a non-empty array containing only positive integers, find if the array can be partitioned into two subsets such that the sum of elements in both subsets is equal.

    Note:

    1. Each of the array element will not exceed 100.
    2. The array size will not exceed 200.

    Example 1:

    Input: [1, 5, 11, 5]
    
    Output: true
    
    Explanation: The array can be partitioned as [1, 5, 5] and [11].
    

    Example 2:

    Input: [1, 2, 3, 5]
    
    Output: false
    
    Explanation: The array cannot be partitioned into equal sum subsets.
    

    解题分析

    能否将数组分成两部分,且两部分彼此相等。

    首先就是将元素都加起来,因为都是正整数,如果和为奇数,那么分成两部分怎么分两部分也不可能相等。

    如果总和为偶数,那么问题就转变成 从数组中,挑出来相加等于总和一半的数,如果挑不出来那就是不存在,否则就是存在。(从一堆物品中挑出来装入背包,背包的总重为 总和的一半 )

    建模:F_k(y) = max\{F_k(y),F_{k-1}(y-w_k)+w_k\}

    F_k(y)表示选前k个数,且总和不超过y时可以加出来的和。当k=n,y=target时,F_k(target)=target那么也就是可以凑出来值为target的组合。若不等,则不存在。

    题解

    public boolean canPartition(int[] nums) {
            int n = nums.length;
            if (n <= 0) return false;
            int sum = 0;
            for (int i = 0; i < n; i++) {
                sum += nums[i];
            }
            if (sum %2 == 1){
                return false;
            }
            int target = sum >> 1;
            int [][]dp = new  int[n+1][target+1];
            //
            int y = target;
            for (int i = 1; i <= n; i++) {
                for (int j = 0; j <=  target; j++) {
                    int left = dp[i-1][j];
                    int right;
                    if (j-nums[i-1] < 0){
                        right = Integer.MIN_VALUE;
                    }
                    else {
                        right = dp[i-1][j-nums[i-1]]+nums[i-1];
                    }
                    dp[i][j] = Math.max(left,right);
                }
            }
            if (dp[n][target] == target)
                return true;
            return false;
        }
    
    

    大神解法

    public boolean canPartition(int[] nums) {
        int sum = 0;
    
        for (int num : nums) {
            sum += num;
        }
    
        if ((sum & 1) == 1) {
            return false;
        }
        sum /= 2;
    
        int n = nums.length;
        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-num];
                }
            }
        }
        return dp[sum];
    }
    

    先说优点,再说其思路。

    代码优点:for的迭代器写法,位操作判断奇偶,备忘录用二进制(节省存储还好用)

    思路:只维护一个一维矩阵,长度为target+1

    递推公式为:F_k(y) = F_k(y) || F(y-w_k)

    F_k(y)表示,使用前k个数,能否凑出和为y。能则为true,不能则为false。子问题还是考y,k来界定的。

    当k=n,y=target时,F_n(target)表示使用所有的数,能否凑出和为target。若能则返回true,否则false。

    相关文章

      网友评论

        本文标题:背包问题(01背包+leetcode416)

        本文链接:https://www.haomeiwen.com/subject/amjsnqtx.html