给你一个 只包含正整数 的 非空 数组 nums
。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
本题需要使用01背包的解题思路
什么事01背包呢,参考下面的链接
0-1背包基础理论一:
文章详解:代码随想录
视频讲解:
带你学透0-1背包问题!| 关于背包问题,你不清楚的地方,这里都讲了!| 动态规划经典问题 | 数据结构与算法_哔哩哔哩_bilibili
0-1背包理论基础二:
文章详解:代码随想录
视频讲解:带你学透01背包问题(滚动数组篇) | 从此对背包问题不再迷茫!_哔哩哔哩_bilibili
代码展示:
- class Solution:
- def canPartition(self, nums: List[int]) -> bool:
- # 使用01背包思路,可以把nums数组中的元素看成物品,物品的重量和价值都是元素的值
- # 使用一维数组,dp[j],表示容量为j的背包最大价值是dp[j]
- # 根据01背包,一维数组的推导公式dp[j] = max(dp[j], dp[j-weight[i]]+value[i])
- # weight[i],value[i]分别表示物品的重量和价值
- # 可以得出本题的推导公式,dp[j] = max(dp[j], dp[j-nums[i]]+nums[i])
- #
- # 要想分割成两个子集,并且元素相等,则nums得为偶数
- # 每个子集的和为nums元素和的一半
- target = sum(nums) # 求nums的和
- if target % 2 != 0:
- return False # 若nums的和不是偶数,则无法分割成两个子集的元素和相等
- target = int(target/2)
- print(target)
- # 初始化
- dp = [0] * (target+1)
- # 遍历顺序
- for i in range(len(nums)):
- for j in range(target, 0, -1):
- if j < nums[i]:
- # 当背包容量小于物品i的重量时,无法放入物品i
- dp[j] = dp[j]
- else:
- # 当背包容量大于等于物品i的重量时,可以放入物品i
- dp[j] = max(dp[j], dp[j-nums[i]]+nums[i])
- print(dp)
- if dp[target] == target:
- # 当最后的背包容量所装的最大价值,等于目标价值
- # 即当子集的元素和,等于目标和,就找到了符合要求的两个子集,返回True
- return True
- else:
- # 不等,说明找不到两个子集的和相等
- return False