LeetCode：698.劃分爲k個相等的子集

題目

給定一個整數數組 nums 和一個正整數 k，找出是否有可能把這個數組分成 k 個非空子集，其總和都相等。

思路

回溯：因爲每個元素都要用上，那取到和爲target的一組值，就設置total爲0重新取。遞歸終止條件是，當沒有值可取且target等於total。一旦找到這樣的集合，提前阻斷，一直返回True
注意：對nums從大到小排序，如果最大的大於target，直接返回False。節省時間。

class Solution:
    def canPartitionKSubsets(self, nums, k):
        target = sum(nums) // k
        if sum(nums) % k :
            return False
        nums.sort(reverse=True)
        print(nums)
        if nums[0] > target:
            return False
        def dfs(remain, total):
            if total == target:
                if len(remain) == 0:
                    return True
                return dfs(remain, 0)
            for i, n in enumerate(remain):
                if (total + n) <= target and dfs(remain[:i]+remain[i+1:], total+n):
                    return True
            return False
        return dfs(nums, 0)