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)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值