1049.最后一块石头的重量II
关键要想到:把石头分成重量尽可能相近的两堆,最后两堆石头的差就是要求的结果。而且,石头 的重量,也是它的价值。
那么后续就和LC 416写法一致了。
class Solution:
def lastStoneWeightII(self, stones: List[int]) -> int:
dp = [0 for _ in range(1501)]
target = sum(stones) // 2
for i in range(len(stones)):
for j in range(target , stones[i] - 1 , -1):
dp[j] = max(dp[j] , dp[j-stones[i]] + stones[i])
return sum(stones) - 2 * dp[target]
JAVA 版本
class Solution {
public int lastStoneWeightII(int[] stones) {
int sum = 0;
int[] dp = new int[1501];
for(int i = 0; i < stones.length ; i ++){
sum += stones[i];
}
int target = sum / 2;
for(int i = 0; i < stones.length; i++){
for(int j = target; j >= stones[i] ; j--){
dp[j] = Math.max(dp[j] , dp[j-stones[i]] + stones[i]);
}
}
return (sum - 2 * dp[target]);
}
}
LeetCode:494.目标和(动规在组合中的应用)
本题思路:
将数组分成两堆,一堆数字将他们添加正号(left),另一堆给他们添加负号(right),那么,有sum(left) + sum(right) = sum(nums),并且sum(left) - sum(right) = target,于是left = (sum + target) / 2。我们找到nums数组中,有几种方法,使取出的数字加起来等于left即可。
抽象为背包问题:
left为背包容量,nums数组中各个数字代表物品重量,我们要找出有几种方法,使得各物品重量之和等于left。
tips:若left = (sum + target) / 2这里不能整除,则说明没有方法,举几个特例即可
dp数组含义:
dp[j]:容量为j的背包,各物品之和等于j的方法有dp[j]种
递推公式:dp[j] += dp[ j - nums[i]],同样,在纸上写几个具体例子即可理解
class Solution:
def findTargetSumWays(self, nums: List[int], target: int) -> int:
total = sum(nums)
if abs(target) > total: return 0
if (total + target) % 2 != 0: return 0
left = (total + target) // 2
dp = [0] * (left + 1)
dp[0] = 1
for i in range(len(nums)):
for j in range(left , nums[i] - 1 , -1):
dp[j] += dp[j - nums[i]]
return dp[left]
JAVA版本
class Solution {
public int findTargetSumWays(int[] nums, int target) {
int sum = 0;
for(int num:nums){
sum += num;
}
if(Math.abs(target) > sum){
return 0;
}
if ((target + sum) % 2 != 0){
return 0 ;
}
int left = (target + sum) / 2;
int[] dp = new int[left + 1];
dp[0] = 1;
for(int i = 0 ; i < nums.length ; i ++){
for(int j = left ; j >= nums[i] ; j --){
dp[j] += dp[j-nums[i]];
}
}
return dp[left];
}
}