1、分割等和子集
题目描述
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
代码
class Solution {
public:
bool canPartition(vector<int>& nums){
int sum=0,flag=0;
int n=nums.size();
//求和
for(int i=0;i<n;i++){
sum+=nums[i];
}
//如果和是奇数,则不可能分割成两个和相等的子集
if(sum%2==1)
return false;
int mid=sum/2;
int dp[mid+1];//注意dp数组开多大
memset(dp,0,sizeof(dp));//初始化
for(int i=0;i<n;i++){
for(int j=mid;j>=nums[i];j--){
dp[j]=max(dp[j],dp[j-nums[i]]+nums[i]);
}
}
if(dp[mid]==mid){
return true;
}
return false;
}
};
总结
这个题不是太难想,按动归五部曲来做:
(1)确定dp下标及dp数组的含义:这个题让求是否能分割成两个和相等的子集,那求mid容量的背包能否被这些数恰好装满就可以了;则dp下标j代表背包容量(1~mid),dp代表这个容量的背包能放的数之和最大是多少;
(2)确定递推公式:dp[j]=max(dp[j],dp[j-nums[i]]+nums[i]);不放当前的数与放当前的数哪个使背包放的数之和大;
(3)初始化:memset(dp,0,sizeof(dp));背包初始化为0;
(4)两层循环:外层遍历物品即nums[],内层遍历背包;随着物品增多,不同容量的背包所能装的物品重量也不断更新,最终求出所有物品能装进背包容量为mid的背包的最大重量;
(5)打印验证(结果不对可以在草稿纸上按递推公式推一下,与打印出来的做对比看看哪里出现了错误)。
题目链接: 416. 分割等和子集
2、最后一块石头的重量 II
题目描述
有一堆石头,用整数数组 stones 表示。其中 stones[i] 表示第 i 块石头的重量。
每一回合,从中选出任意两块石头,然后将它们一起粉碎。假设石头的重量分别为 x 和 y,且 x <= y。那么粉碎的可能结果如下:
如果 x == y,那么两块石头都会被完全粉碎;
如果 x != y,那么重量为 x 的石头将会完全粉碎,而重量为 y 的石头新重量为 y-x。
最后,最多只会剩下一块 石头。返回此石头 最小的可能重量 。如果没有石头剩下,就返回 0。
示例 1:
输入:stones = [2,7,4,1,8,1]
输出:1
解释:
组合 2 和 4,得到 2,所以数组转化为 [2,7,1,8,1],
组合 7 和 8,得到 1,所以数组转化为 [2,1,1,1],
组合 2 和 1,得到 1,所以数组转化为 [1,1,1],
组合 1 和 1,得到 0,所以数组转化为 [1],这就是最优值。
示例 2:
输入:stones = [31,26,33,21,40]
输出:5
提示:
1 <= stones.length <= 30
1 <= stones[i] <= 100
代码
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 mid=sum/2;
int dp[mid+1];//注意dp下标范围
memset(dp,0,sizeof(dp));
for(int i=0;i<n;i++){//i:物品
for(int j=mid;j>=stones[i];j--){
//j:背包容量,dp[j]容量j的背包最多盛放物品的重量
dp[j]=max(dp[j],dp[j-stones[i]]+stones[i]);
}
}
int surplus=sum-2*dp[mid];
return surplus;
}
};
总结
这个题比上面的题难想很多,被卡了好久,然后看题解的思路,豁然开朗…
这个题可以用归纳法(如下图),转换成把石头分为两堆,求这两堆石头重量之差的最小值,这两堆石头的重量越接近mid,它们重量之差就越小。即求背包容量为mid的背包最多能装多少重量的物品,然后再求两堆物品的差值就可以了。
无论两堆中石头重量是如何,经过互相减之后总能算出两堆重量之差。
题目链接: 1049. 最后一块石头的重量 II
附上题解: 官方题解