k个即确定目标target = sum/k
target非整数 false
nums中有大于target的数
都不大于顺便先把等于的去掉k--
创建一个新的新k大小空数组,回溯每次加入一个nums,如果加入任何一个nums失败就false
因为本身就是sum除以k的target,能全部塞进去就是成功了
public class 划分为k个相等的子集 {
public static void main(String[] args) {
划分为k个相等的子集 a = new 划分为k个相等的子集();
int[] nums = {4, 3, 2, 3, 5, 2, 1};
System.out.println(a.canPartitionKSubsets(nums,4));
}
public boolean canPartitionKSubsets(int[] nums, int k) {
int sum = Arrays.stream(nums).sum();
if (sum % k > 0) return false;
//由题意我们可以知道目标值绝对是sum/k
int target = sum / k;
//给nums数组排序,递增顺序
Arrays.sort(nums);
int row = nums.length - 1;
//如果最大的值大于目标值,则改数组肯定不满足条件,直接返回false
if (nums[row] > target) return false;
//如果有等于目标值的,k--,row--
while (row >= 0 && nums[row] == target) {
row--;
k--;
}
return search(new int[k], row, nums, target);
}
//函数作用:搜索nums中的每一个数,能不能添加到groups里,并且和等于target,row代表当前添加到groups中的数
public boolean search(int[] groups, int row, int[] nums, int target) {
//如果nums中没有数了,则代表所有数全部放入groups中
if (row < 0) return true;
//注意row--,先赋值,再--
int v = nums[row--];
//遍历groups的k个子集
for (int i = 0; i < groups.length; i++) {
if (groups[i] + v <= target) {
groups[i] += v;
if (search(groups, row, nums, target)) return true;
//不能添加到groups中,则把它减去,及回溯
groups[i] -= v;
}
//如果遍历完了,所有数都不能放进groups中,及没有值相加或者本身等于target,则跳出循环break,或者return false;
if (groups[i] == 0) return false;
}
return false;
}
}