LeetCode 698.划分为k个相等的子集

给定一个整数数组 nums 和一个正整数 k，找出是否有可能把这个数组分成 k 个非空子集，其总和都相等。

示例 1：

输入： nums = [4, 3, 2, 3, 5, 2, 1], k = 4
输出： True
说明： 有可能将其分成 4 个子集（5），（1,4），（2,3），（2,3）等于总和。

提示：

1 <= k <= len(nums) <= 16
0 < nums[i] < 10000

class Solution:
    def canPartitionKSubsets(self, nums: List[int], k: int) -> bool:
        target, rem = divmod(sum(nums), k)
        if rem!=0:
            return False

        nums = sorted(nums)   ## 先进行排序
        print(nums)
        cur = [0]*k
        return self.canPartitionKSubsets_helper(nums,cur,target)

    def canPartitionKSubsets_helper(self,nums,cur,target):
        if len(nums) == 0: ## 所有数都安放完成，返回True
            return True
        num = nums.pop()   ## 拿出最大值开始搜索
        for i, value in enumerate(cur):
            if value + num <= target:   ##　如果满足要求递归进行搜索
                cur[i]+=num
                if self.canPartitionKSubsets_helper(nums,cur,target):
                    return True 
                cur[i]-=num             ## 递归返回后恢复上一层状态 
            if value == 0:             ## 尽量将0放在最后进行搜索
                break
        nums.append(num)
        return False