其他分享
首页 > 其他分享> > leetcode 698 集合k划分

leetcode 698 集合k划分

作者:互联网

集合划分

参考:负雪明烛

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

面试的时候面到这一题,大脑一片空白。知道大致的方向,深搜,回溯。但是就是不知道如何下手,练的太少。

class Solution:
    def canPartitionKSubsets(self, nums: List[int], k: int) -> bool:
        if not nums or len(nums) < k:
            return False
        sum_ = sum(nums)
        div, mod = divmod(sum_,k)
        if sum_%k and max(nums)>sum_/k:
            # 注意这里的判断条件
            return False
        # 每个集合的和都是一致的
        nums.sort(reverse=True)
        target = [div]*k 
        return self.dfs(nums,k,0,target)

    def dfs(self,nums,k,index,target):
        # 如果恰好用完了,说明满足条件 
        if index == len(nums):
            return True
        num = nums[index]
        for i in range(k):
            # 保证target最终都是0,即所有数字都到了对的地方
            if target[i]>=num:
                target[i] -= num
                if self.dfs(nums,k,index+1,target):
                    return True
                target[i] += num
            # 可以将这个过程看作入栈的过程
        return False

标签:index,return,target,nums,698,self,sum,集合,leetcode
来源: https://www.cnblogs.com/curtisxiao/p/11207352.html