leetcode 698、划分为k个相等的子集(递归)

该博客探讨了LeetCode第698题,即如何将给定数组分成k个和相等的非空子集。通过递归方法,首先判断数组总和是否能被k整除,然后使用一个mark数组记录已选择的元素,并递归地尝试构建和为子集总和的子集。当找到k个满足条件的子集时返回true,否则返回false。
摘要由CSDN通过智能技术生成

给定一个整数数组 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

思路:

这道题我们可以用递归来做。

首先我们求出数组的所有数字的和 sum,判断 sum 是否能整除 k,不能整除的话直接返回 false。

然后需要一个 mark 数组来记录哪些数组已经被选中了,然后调用递归函数,我们的目标是组k个子集合,每个子集合的和为 subSum = sum / k。我们还需要变量 start,表示从数组的某个位置开始查找,curSum 为当前子集合的和。

在递归函数中,如果 k = 1,说明此时只需要组一个子集合,那么当前的就是了,直接返回 true。

如果 curSum 等于 subSum 了,那么我们再次调用递归,此时传入 k - 1,start 和 curSum 都重置为0,因为我们当前又找到了一个和为 target 的子集合,要开始继续找下一个。
否则的话就从 start 开始遍历数组,如果当前数字已经访问过了则直接跳过,否则标记为已访问。

然后调用递归函数,k 保持不变,因为还在累加当前的子集合,start 传入 i + 1,curSum 传入 curSum + nums[i],因为要累加当前的数字,如果递归函数返回true了,则直接返回 true。
否则就将当前数字重置为未访问的状态继续遍历。


/*
* @param sum    数组总和
* @param mark[] 记录每个元素是否被标记过
* @param start  每个子数组的起始元素下标
* @param subSum 子数组的总和
* @param curSum 当前子数组的和
*/
class Solution {
   
    publi
  • 5
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值