416 分割等和子集
题目很难,背包问题递推式子也比较难理解。。
看了两期视频才理解了
关键点有二:
一是递推公式
二是遍历顺序,特别是当把二维dp压缩为一维的时候必须先遍历物品再遍历容量,且内层循环要倒序遍历,从最大容量到物品i的重量(小于物品i的重量就放不进去i了)
class Solution:
def canPartition(self, nums: List[int]) -> bool:
s = sum(nums)
if s%2 == 1:
return False
else:
target = int(s/2)
dp = [0]*10001
for i in range(len(nums)):
for j in range(target,nums[i]-1,-1):
dp[j] = max(dp[j],dp[j-nums[i]]+nums[i])
print(dp)
if dp[target] == target: return True
return False
1049 最后一块石头重量2
没有理解到这题本质是把石头分成重量尽可能相近的两堆的本质,看解答知道这一点后就很好写了
class Solution:
def lastStoneWeightII(self, stones: List[int]) -> int:
s = sum(stones)
dp = [0]*(s+1)
for i in range(len(stones)):
for j in range(s,stones[i]-1,-1):
dp[j] = max(dp[j],dp[j-stones[i]]+stones[i])
# print(dp)
res = s
for i in range(s+1):
res = min(res,abs(s-2*dp[i]))
return res
494 目标和
很难,视频都看了两遍才大概弄明白怎么做的
首先要分析出来需要把nums分成两个集合,一个集合的和需要达到(sum+target)/2
然后再使用dp分析有多少种方法满足这样的背包装的方式
递推式dp[j] += dp[j-nums[i]]
并且要注意这种装包方式数量的问题初始值dp[0]为1
class Solution:
def findTargetSumWays(self, nums: List[int], target: int) -> int:
s = sum(nums)
if abs(target) > s:return 0
if (target+s)%2 == 1: return 0
bagSize = int((target+s)/2)
dp = [0]*(bagSize+1)
dp[0] = 1
for i in range(len(nums)):
for j in range(bagSize,nums[i]-1,-1):
dp[j] += dp[j-nums[i]]
return dp[-1]
474 一和零
题目思路不难,一个有二维容量的背包问题
class Solution:
def findMaxForm(self, strs: List[str], m: int, n: int) -> int:
dp = [[0]*(n+1) for i in range(m+1)]
# print(dp)
for i in range(len(strs)):
num0, num1 = self.cal(strs[i])
for j in range(m,num0-1,-1):
for k in range(n,num1-1,-1):
# print(j,k,num0,num1)
dp[j][k] = max(dp[j][k],dp[j-num0][k-num1]+1)
print(dp)
return dp[m][n]
def cal(self,string):
num1 = 0
num0 = 0
for i in string:
if i == '1':
num1 += 1
else:
num0 += 1
return num0,num1
518 零钱兑换2
完全背包,算装包方式数量
class Solution:
def change(self, amount: int, coins: List[int]) -> int:
dp = [0]*(amount+1)
dp[0] = 1
for i in range(len(coins)):
for j in range(coins[i],amount+1):
dp[j] += dp[j-coins[i]]
print(dp)
return dp[-1]
377 组合总和IV
完全背包,算装包方式数量,然后对顺序排列组合
重点在于dp数组中只记录了组合方式,如何获取到具体每个组合能否更改顺序?
如果用01背包,可以找到组合中每个元素都不同的组合方式
看了解答发现居然只要交换循环顺序就可以了??!为什么这么神奇
看了讲解视频大概明白了,完全背包本身无所谓遍历顺序,但是当记录的不是装包值而是装包方式的时候,先遍历物品再遍历背包容量dp是组合数,不考虑顺序,也就是更换顺序视作同一个方式;而先遍历背包容量再遍历物品则dp是排列数,考虑了不同顺序产生的不同装包方式。
class Solution:
def combinationSum4(self, nums: List[int], target: int) -> int:
dp = [0]*(target+1)
dp[0] = 1
for j in range(target+1):
for i in range(len(nums)):
if j-nums[i] >= 0:
dp[j] += dp[j-nums[i]]
print(dp)
return dp[-1]