class Solution(object):
def canPartition(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
# 1.dp 用01背包的思想去解决这个问题
# 针对这个题,我们要把数组nums分割成两个子集,使得两个子集的元素和相等
# 那么我对数组nums求和,然后/2,如果可以整除说明可能会分割成两个元素和相等的子集
# 如果不能整除2,说明一定不可能分成两个元素和相等的子集
# dp[j]代表背包的容量。其中j=target=sum/2是我们的目标
# 01背包问题的状态转移方程为:dp[i][j] = max(dp[i-1][j], dp[i-1][j-weight[i]+value[i]])
# 改进版:dp[j] = max(dp[j], dp[j-weight[i]+value[i]])
# nums的数值即时他所占的大小,也是他的价值
n = len(nums)
if n < 2:
return False
sum_ = sum(nums)
if sum_ % 2 == 1:
return False
target = sum_ / 2
dp = [0 for _ in range(target+1)]
# 先遍历物品,再遍历背包
# 一维的遍历背包需要倒序,避免物品重复
# nums[i]-1:背包大小最小的值,-1是因为[)
for i in range(n):
for j in range(target, nums[i]-1, -1):
dp[j] = max(dp[j], dp[j-nums[i]]+nums[i])
return dp[target] == target
动态规划--LC416分割等和子集
最新推荐文章于 2024-07-25 19:34:38 发布