题目
(473. 火柴拼正方形也是这一类型的,划分为四个子集的情况)
解题思路
题416是题698(k=2时)的特例。
划分为k个相等的子集的思路:
直接DFS判断是否能够划分成为k个子集,重点在于剪枝,即需要将数组从大到小进行排序。为什么不是从小到大排序?也是可以的,但是(在测试用例中)从大到小可以减少回溯的次数,大值+大值+… >= 目标值的几率更大。
当只划分为了两个集合时,可以直接套用上面划分为k个相等子集的代码,也可以利用01背包动态规划。动态规划思路:只要当前数组中能填充背包容量为sum(nums) // 2,那就可以划分为两个。填充背包相当于我们已经找到了一个子集,剩下的没有填充到背包中的元素总和肯定 = sum(nums) // 2,相当于另一个子集。
代码实现
"""
698. 划分为k个相等的子集
"""
class Solution:
def canPartitionKSubsets(self, nums, k: int) -> bool:
# 从大到小排序
nums = sorted(nums, reverse=True)
# 每个集合的目标总和
sums = sum(nums) // k
if sums < max(nums) or sum(nums) % k:
return False
vis =