代码随想录–动态规划部分
day 36 动态规划第四天
一、力扣1049–最后一块石头的重量Ⅱ
代码随想录题目链接:代码随想录
有一堆石头,用整数数组 stones 表示。其中 stones[i] 表示第 i 块石头的重量。
每一回合,从中选出任意两块石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下,就返回 0。
考虑dp应该记录什么
这个题可以转换为:分成两堆石头,使他们重量之差最小
就成为了01背包问题
那么dp就是背包的容量,其上限是石头重量总和的一半,因为是要分出来重量总和最相近的两堆石头
物品的重量呢?就是stone[i],同时物品的价值也是stone[i]
则可以得到递推公式dp[j] = max( dp[j], dp[j - stone[i]] + stone[i] )
再考虑dp数组的初始化问题
因为重量肯定是大于0的,那么max时只要保证初始化的数字能够被物品价值覆盖就行,所以设置为0没问题
遍历顺序也就出来了,外循环是循环物品,内循环是dp从后向前更新
代码如下:
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
vector<int> dp(15001, 0);
int sum = 0;
for (int i = 0; i < stones.size(); i++) sum += stones[i];
int target = sum / 2;
for(int i = 0 ; i < stones.size(); i ++)
for(int j = target; j >= stones[i]; j --)
dp[j] = max(dp[j], dp[j-stones[i]]+ stones[i]);
return sum - 2 * dp[target];
}
};
二、力扣494–目标和
代码随想录题目链接:代码随想录
给你一个非负整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个 表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
假设前面是+号的数字之和为x,那么前面是-号的数字之和就是sum-x
所以我们要求的是 x - (sum - x) = target的情况
x = (target + sum) / 2,这就是01背包的最大容量了,实际上就是要求一个问题:
从num中取数字相加,使他们的和为x,有多少种方法
所以dp[j]=0,1,… x,代表背包容量递增
递推公式就是dp[j] = dp[j - num[i]]
意思是差一个num[i]就能凑整dp[j]时,dp[j]的填满方法肯定是有dp[j-num[i]]种的
代码如下:
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int sum = 0;
for (int i = 0; i < nums.size(); i++) sum += nums[i];
if (abs(target) > sum) return 0;
if ((target + sum) % 2 == 1) return 0;
int bagSize = (target + sum) / 2;
vector<int> dp(bagSize + 1, 0);
dp[0] = 1;
for (int i = 0; i < nums.size(); i++)
for (int j = bagSize; j >= nums[i]; j--)
dp[j] += dp[j - nums[i]];
return dp[bagSize];
}
};
三、力扣474–一和零
代码随想录题目链接:代码随想录
给你一个二进制字符串数组 strs 和两个整数 m 和 n 。
请你找出并返回 strs 的最大子集的长度,该子集中 最多 有 m 个 0 和 n 个 1 。
定义:如果 x 的所有元素也是 y 的元素,集合 x 是集合 y 的 子集 。
动态数组定义为dp[i][j]:有i个0和j个1的最大子集长度
将这个数组扩充完,取右下角的值输出就行
怎么递推?这依然是01背包问题,因为每个元素只能取一次
背包内的物品就是 s t r ∈ s t r s str\in strs str∈strs的每个字符串
那么每个外循环都会增加一个str的选项,其影响方式是0和1的个数
所以dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1)
代码如下:
class Solution {
public:
int findMaxForm(vector<string>& strs, int m, int n) {
vector<vector<int>> dp(m + 1, vector<int>(n + 1,0));
for(auto ele : strs)
{
int zeroNum = 0;
int oneNum = 0;
for(auto c : ele)
{
if(c == '0') zeroNum ++;
else oneNum ++;
}
for(int i = m; i >= zeroNum; i --)
for(int j = n; j >= oneNum; j --)
dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
}
return dp[m][n];
}
};