partition-to-equal-k-subset
Input: nums = [4,3,2,3,5,2,1], k = 4
Output: true
Explanation: It's possible to divide it into 4 subsets (5), (1, 4), (2,3), (2,3) with equal sums.Input: nums = [1,2,3,4], k = 3
Output: falseclass Solution:
def canPartitionKSubsets(self, nums: List[int], k: int) -> bool:
total = sum(nums)
if total%k:
return False
target=total//k
length=len(nums)
used=[False]*length
def Helper(i,k,currentSum):
if k==0:
return True
if currentSum==target:
return Helper(0,k-1,0)
for j in range(i,length):
newCurrentSum=currentSum+nums[j]
if used[j] or newCurrentSum>target:
continue
used[j]=True
if Helper(j,k,newCurrentSum):
return True
used[j]=False
return False
return Helper(0 , k, 0)Last updated