leetcode698-划分为k个相等的子集

标签：arr 相等 return target nums int sum 子集 leetcode698

划分为k个相等的子集

• 回溯+剪枝

首先先判断总和sum能否被整除。然后对数组排序，从后向前遍历。

1. 如果当前的值大于target，表明最大值已经超出范围，直接返回false
2. 如果当前的值等于target，那么直接将其放入bucket中
3. 如果当前的值小于target，需要进行回溯，从当前位置向前回溯即可

回溯流程：对于每个index，判断放入bucket之后会不会超出target范围，如果不会，则将其放入，开启下一轮回溯；否则直接跳过。

class Solution {
    int arr[];
    public boolean canPartitionKSubsets(int[] nums, int k) {
        int sum = 0, n = nums.length;
        for(int val : nums) sum += val;
        if(sum % k != 0)    return false;
        Arrays.sort(nums);
        arr = new int[k];
        for(int i = n-1, cur = k-1, target = sum / k; i >= 0; i--){
            if(nums[i] > target)    return false;
            else if(nums[i] == target)  arr[cur--] = target;
            else    return backtracking(nums, i, target);
        }
        return true;
    }
    public boolean backtracking(int[] nums, int k, int target){
        if(k == -1) return true;
        for(int i = 0; i < arr.length; i++){
            if(arr[i] + nums[k] <= target){
                arr[i] += nums[k];
                if(backtracking(nums, k-1, target)) return true;
                arr[i] -= nums[k];
            }
        }
        return false;
    }
}

标签：arr,相等,return,target,nums,int,sum,子集,leetcode698
From： https://www.cnblogs.com/xzh-yyds/p/16588564.html