01背包问题做题总结(一)

01背包问题做题总结(一),第1张

01背包问题做题总结(一) 这周是两道类似的题

分割等和子集
题目概述

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

class Solution {
    public boolean canPartition(int[] nums) {
        //dp[j]:背包容量为j,本体j指目标sum/2,dp[j]为最大可以组成j的自己总和
        //weight为nums[i],value为nums[i]
        //递推公式为dp[j] = Math.max(dp[j],dp[j - nums[i]] + nums[i]);
        int sum = 0;
        for(int i = 0;i < nums.length;i++) {
            sum += nums[i];
        }
        if(sum % 2 == 1) {
            return false;
        }
        int tar = sum / 2;
        int[] dp = new int[tar + 1];
        dp[0] = 0;
        for(int i = 0;i < nums.length;i++) {
            for(int j = tar;j >= nums[i];j--) {
                dp[j] = Math.max(dp[j],dp[j - nums[i]] + nums[i]);
            }
        }
        return dp[tar] == tar; 
    }
}

最后一块石头的重量2

class Solution {
    public int lastStoneWeightII(int[] stones) {
        //这道题意思其实就是把这堆石头分成尽可能相同的两堆,用大的减小的就是最小的石头重量
        //那么dp定义为:放入容量(重量)为j的背包中,背包的最大重量
        //递推公式:dp[j] = Math.max(dp[j],dp[j - stones[i]] + stones[i]);
        //初始化:stone的最大重量为30 * 100 ,那一半就是15000,dp开这么大就够了
        //重量都是正整数,所以初始化为0即可
        int[] dp = new int[150000];
        dp[0] = 0;
        int sum = 0;
        for(int i = 0;i < stones.length;i++) {sum += stones[i];}
        int tar = sum / 2;
        for(int i = 0;i < stones.length;i++) {
            for(int j = tar;j >= stones[i];j--) {
                dp[j] = Math.max(dp[j],dp[j - stones[i]] + stones[i]);
            }
        }
      
        //tar向下取整,所以重量小于一半,dp[tar] < sum - dp[tar]
        return (sum - dp[tar]) - dp[tar]; 
    }
}

做背包的题,我发现最重要的是对题意的理解,然后想办法把它转换为可以套01背包模板的形式,这是最难的,感觉还需要再练练。

语文不好

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5703051.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存