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;
- }
- }