0
点赞
收藏
分享

微信扫一扫

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

橙子好吃吗 2022-01-12 阅读 52

这周是两道类似的题

分割等和子集
题目概述

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背包模板的形式,这是最难的,感觉还需要再练练。

举报

相关推荐

0 条评论