● 416. 分割等和子集
class Solution:
def canPartition(self, nums: List[int]) -> bool:
total = sum(nums)
target = total // 2
if total % 2 != 0:
return False
dp = [0] * (target + 1)
for num in nums:
for j in range(target, num - 1, -1): num-1即当物品比背包还重时,肯定放不了,背包就不需要遍历了。
dp[j] = max(dp[j], dp[j - num] + num)
if dp[target] == target: # 这个地方跟卡哥写的不太一样,卡哥放在for外面了,这样也对
return True
return False
【思考】本质是01背包问题,每个num只能放一次。例:[5,1,11,5],dp[j] = max(dp[j], dp[j - num] + num)的意义:
dp[j]: 取上物品nums[0]~nums[num]时,当背包容量为j时能装的价值。这里要理解:1. 背包容量取值是for j in range(target, num - 1, -1),也就是从11每次-1取到num。 2.比如num = 5,dp[5],dp[6]~dp[11]的value都是5。
dp[j - num] + num :不装当前物品num时,背包的价值,加上物品num本身的价值。
用一维数组不断更新dp[j]( j 范围是11-num),最后如果有dp[11] = 11的时候,那么就是True了。