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