美文网首页动态规划
背包问题(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