Leetcode 1049.最后一块石头的重量II
题目链接:Leetcode 1049.最后一块石头的重量II
题目描述: 有一堆石头,用整数数组 stones
表示。其中 stones[i]
表示第 i
块石头的重量。
每一回合,从中选出任意两块石头,然后将它们一起粉碎。假设石头的重量分别为 x
和 y
,且 x <= y
。那么粉碎的可能结果如下:
- 如果
x == y
,那么两块石头都会被完全粉碎; - 如果
x != y
,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为y-x
。
最后,最多只会剩下一块石头。返回此石头最小的可能重量 。如果没有石头剩下,就返回 0
。
思路: 本题可以利用01背包来解决。为何会想到利用01背包呢?我们的目的是求剩余重量的最小值,那么对于所有石头,我们需要尽量让石头分成重量相同的两堆,这样相撞之后剩下的石头最小。然后思路与这道题类似416. 分割等和子集,只不过最后结果需要返回两个子集合的差值。
代码如下:
class Solution {
public:
int lastStoneWeightII(vector<int>& stones) {
int dp[15010] = {0};
int sum = accumulate(stones.begin(), stones.end(), 0);
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 - dp[target]) - dp[target];
}
};
- 时间复杂度:
O
(
m
×
n
)
O(m × n)
O(m×n) ,
m
是石头总重量(准确的说是总重量的一半),n
为石头块数 - 空间复杂度: O ( m ) O(m) O(m)
Leetcode 494.目标和
题目链接:Leetcode 494.目标和
题目描述: 给你一个非负整数数组 nums
和一个整数 target
。
向数组中的每个整数前添加 +
或 -
,然后串联起所有整数,可以构造一个 表达式 :
- 例如,
nums = [2, 1]
,可以在2
之前添加+
,在1
之前添加-
,然后串联起来得到表达式"+2-1"
。
返回可以通过上述方法构造的、运算结果等于 target
的不同表达式的数目。
思路: 本题可以将数组分为两个集合,分别代表正数集合和负数集合,然后利用01背包求解。假设加法的总和为x
,那么减法对应的总和就是sum - x
。所以我们要求的是 x - (sum - x) = target
解得x = (target + sum) / 2
,此时问题就转化为,装满容量为x
的背包有几种方法。
代码如下:
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int sum = accumulate(nums.begin(), nums.end(), 0); //数组元素总和
//经过计算,转化为填满容量为(sum + target) / 2 的背包
if (abs(target) > sum) //目标值比总和还大
return 0;
if ((target + sum) & 1) //相加为奇数,则无方案
return 0;
int x = (sum + target) / 2;
vector<int> dp(x + 5, 0);
dp[0] = 1;
for (int i = 0; i < nums.size(); i++)//遍历nums数组
for (int j = x; j >= nums[i]; j--)//遍历背包容量
dp[j] += dp[j - nums[i]];
return dp[x];
}
};
- 时间复杂度: O ( n × m ) O(n × m) O(n×m)
- 空间复杂度: O ( m ) O(m) O(m)
Leetcode 474.一和零
题目链接:Leetcode 474.一和零
题目描述: 给你一个二进制字符串数组 strs
和两个整数 m
和 n
。
请你找出并返回 strs
的最大子集的长度,该子集中 最多 有 m
个 0
和 n
个 1
。
如果 x
的所有元素也是 y
的元素,集合 x
是集合 y
的 子集 。
思路: 本题也是01背包的问题,只不过每件物品具有两种性质:0
的数量和1
的数量
代码如下:
class Solution {
public:
int findMaxForm(vector<string>& strs, int m, int n) {
int dp[m + 5][n + 5];
memset(dp, 0, sizeof(dp));
for (auto str : strs) { //遍历集合
//统计某个字符串的0和1数目
int c0 = count(str.begin(), str.end(), '0'), c1 = str.size() - c0;
//倒序遍历求dp数组
for (int i = m; i >= c0; i--)
for (int j = n; j >= c1; j--)
dp[i][j] = max(dp[i][j], dp[i - c0][j - c1] + 1);
}
return dp[m][n];
}
};
- 时间复杂度:
O
(
k
m
n
)
O(kmn)
O(kmn),
k
为strs
的长度 - 空间复杂度: O ( m n ) O(mn) O(mn)
总结
背包问题的难点在于如何将题目的条件抽象成背包模型,这里简单对01背包做个总结之后方便复习:
- Leetcode 416. 分割等和子集是求:给定背包容量,能不能装满这个背包。
- Leetcode 1049.最后一块石头的重量II是求:给定背包容量,尽可能装,最多能装多少
- Leetcode 494.目标和是求:给定背包容量,装满背包有多少种方法。
- Leetcode 474.一和零是求:给定背包容量,装满背包最多有多少个物品。
最后,如果文章有错误,请在评论区或私信指出,让我们共同进步!