Bootstrap

【Day31 LeetCode】动态规划DP Ⅳ

一、动态规划DP Ⅳ

1、最后一块石头的重量II 1049

这题有点像脑筋急转弯,尽量让石头分成重量相同的两堆(尽可能相同),相撞之后剩下的石头就是最小的。明白这一点,就与上一篇博客里的划分等和数组很相似。划分等和数组是给定背包容量,能不能恰好填满该背包;这题是给定背包容量,尽可能填满该背包。直接套用代码。

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int ss = accumulate(stones.begin(), stones.end(), 0);
        int s = ss / 2;
        vector<int> dp(s + 1);
        for(int stone : stones)
            for(int j=s; j>=stone; --j)
                dp[j] = max(dp[j], dp[j-stone] + stone);
        return ss - 2 * dp[s];
    }
};

2、目标和 49

这题需要变通一下,本质上是将原数组分成两个子集,记为left(表示+)和right(表示-),两个子集需要满足: left = (target + sum)/2 。 left组合 - right组合 = target,left + right = sum,而sum是固定的,left - (sum - left) = target 推导出 left = (target + sum)/2 。与上一篇博客里的划分等和数组很相似。此时问题变成了 从nums数组中选取元素填满容量为left的背包的方法。这时套用01背包一维数组的代码,需要修改dp方程。对于二维数组,dp[i][j]表示在0~i中选取元素构成和为j的组合的个数,当前值dp[i][j]有选与不选物品i两个选择,所以递推方程为 d p [ i ] [ j ] = d p [ i − 1 ] [ j ] + d p [ i − 1 ] [ j − n u m s [ i ] ] dp[i][j] = dp[i-1][j] + dp[i-1][j-nums[i]] dp[i][j]=dp[i1][j]+dp[i1][jnums[i]],相应的一维为 d p [ j ] + = d p [ j − n u m s [ i ] ] dp[j] += dp[j-nums[i]] dp[j]+=dp[jnums[i]]

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int s = accumulate(nums.begin(), nums.end(), 0);
        if(abs(target) > s || (target + s) % 2 == 1)
            return 0;
        s = (s + target) / 2;
        vector<int> dp(s + 1);
        dp[0] = 1;
        for(int i=0; i<nums.size(); ++i)
            for(int j=s; j>=nums[i]; --j)
                dp[j] += dp[j-nums[i]]; 
        return dp[s];
    }
};

3、一和零 474

这题是给定背包容量,求装满背包最多有多少物品,并且该背包很特殊,有0和1的数量两个维度。套用优化掉物品维度的01背包代码,dp[i][j]表示最多有i个0和j个1的strs的最大子集的大小为dp[i][j],这里采用二维数组表示背包的维度,物品的维度呗优化掉了,所以在遍历背包时需要和之前一样采用逆序遍历

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int>> dp(m+1, vector<int>(n+1));
        for(string str : strs){  // 遍历物品
            int one = 0, zero = 0;
            for(char ss : str){
                if(ss=='1')
                    ++one;
                else
                    ++zero;
            }
            // 遍历背包
            for(int i=m; i>=zero; --i)
                for(int j=n; j>=one; --j)
                    dp[i][j] = max(dp[i-zero][j-one] + 1, dp[i][j]);
        }
        return dp[m][n];
    }
};

二、写在后面

难点在于将问题分析清楚,理清如何转换成背包问题。第一题是给定背包容量,尽可能装,最多能装多少;第二题是给定背包容量,求装满背包的方法;第三题是给定背包容量,求装满背包最多有多少物品,并且此背包比较特殊,有两个维度。

;