代码随想录算法训练营
Day36 代码随想录算法训练营第 36 天 |LeetCode1049. 最后一块石头的重量 II LeetCode 494. 目标和 LeetCode 474.一和零
目录
前言
LeetCode1049. 最后一块石头的重量 II
LeetCode 494. 目标和
LeetCode 474.一和零
一、LeetCode1049. 最后一块石头的重量 II
1.题目链接
2.思路
(1)尽量让石头分成重量相同的两堆,相撞之后剩下的石头最小,这样就化解成01背包问题---->尽可能接近一半—>在总质量小于等于质量和/2的情况下,让质量尽可能大,得到一堆的质量,sum-dp结果为另外一堆的质量
(2)01背包问题
背包质量上限:target(质量和的一半)
物品价值:stones[i]
物品质量:stones[i]
3.题解
二维
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
int n=stones.size();
int sum=0;
for(int i=0;i<n;i++)
{
sum+=stones[i];
}
vector<vector<int>>dp(n+1,vector<int>(sum,0));
int target=sum/2;
for(int i=stones[0];i<=target;i++)
{
dp[0][i]=stones[0];
}
for(int i=1;i<n;i++)
{
for(int j=0;j<=target;j++)
{
if(j<stones[i])dp[i][j]=dp[i-1][j];
else dp[i][j]=max(dp[i-1][j],dp[i-1][j-stones[i]]+stones[i]);
}
}
return sum-dp[n-1][target]-dp[n-1][target];
}
};
一维
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
int n = stones.size();
int sum = 0;
for (int i = 0; i < n; i++) {
sum += stones[i];
}
int target = sum / 2;
vector<int> dp(target + 5, 0);
for (int i = 0; i < n; i++) {
for (int j = target; j >= stones[i]; j--) {
dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]);
}
}
return sum - dp[target] - dp[target];
}
};
二、LeetCode 494. 目标和
1.题目链接
2.思路
(1)设前面取+的元素之和为x,前面取-的元素之和为sum-x。target=x-(sum-x),解出来x=(target+sum)/2
其中,有两个情况无解:
1)target的绝对值大于sum;
2)target+sum为奇数
(2)01背包问题
1)物品质量:nums[i]
物品价值:nums[i]
包的最大容积:X
dp[j]表示包里面物品质量和为J时,有几种方法
2)状态转移
使得包里面物品质量和为J有n种可能性(可能是加上了nums[0]~nums[n-1]使得总质量j),所以方法数是这些可能性方法数总和
3)初始化
dp[0]=1:在质量为0时,只有不装这一种办法
4)遍历:一维01背包的遍历法
i从小到大,j从大到小
3.题解
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int n=nums.size();
int sum=0;
for(int i=0;i<n;i++)
{
sum+=nums[i];
}
if(abs(target)>sum || (sum+target)%2==1)return 0;
int x=(sum+target)/2;
vector<int>dp(x+5,0);
dp[0]=1;
for(int i=0;i<n;i++)
{
for(int j=x;j>=nums[i];j--)
{
dp[j]+=dp[j-nums[i]];
}
}
return dp[x];
}
};
三、LeetCode 474.一和零
1.题目链接
2.思路
(1)看成有两层的0-1背包,m和n本质上是背包两个层各自的最大容量
- 容量为m的层存放1,容量为n的层存放0
- 物品质量:1(一个字符串为一个物品)
- 物品重量:0,1个数
(2)dp[i][j]:背包两层重量分别为i,j时,最大字符串数
(3)遍历
1)遍历物品:遍历数组中的字符串
2)遍历容量:遍历m和n,从后向前
3.题解
class Solution {
public:
int findMaxForm(vector<string>& strs, int m, int n) {
vector<vector<int>> dp(n + 5, vector<int>(m + 5, 0));
int s = strs.size();
// mn是背包的两个容量,可以看成背包有两层
dp[0][0] = 0;
for (int r = 0; r < s; r++) {
int min0 = 0;
int min1 = 0;
for (int k = 0; k < strs[r].size(); k++) {
if (strs[r][k] == '0')
min0++;
else
min1++;
}
for (int i = n; i >= min1; i--) {
for (int j = m; j >= min0; j--) {
dp[i][j]=max(dp[i][j],dp[i-min1][j-min0]+1);
}
}
}
return dp[n][m];
}
};