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

1.题目描述：

给定一个整数数组nums和一个正整数k，找出是否有可能把这个数组分成 k个非空子集，其总和都相等。

2.回溯：思路见leetcode题解。

class Solution {
    public boolean canPartitionKSubsets(int[] nums, int k) {
        int sum = 0;
        for (int i = 0; i < nums.length; i++) sum += nums[i];
        if (sum % k != 0) return false;//剪枝
        int target = sum / k;
        Arrays.sort(nums);//剪枝，贪心思路，排序后倒序遍历
        if (nums[nums.length - 1] > target) return false;
        boolean[] flag = new boolean[nums.length];
        return backTracking(nums, k, target, nums.length - 1, 0, flag);
    }

    public boolean backTracking(int[] nums, int k, int target, int index, int curSum, boolean[] flag) {
        if (k == 1) return true;//剪枝，k-1个组合满足条件即可
        if (curSum == target) return backTracking(nums, k - 1, target, nums.length - 1, 0, flag);//找到其中一个组合，再重新找
        for (int i = index; i >= 0; i--) {
            if (flag[i] || curSum + nums[i] > target) continue;//剪枝
            flag[i] = true;
            if (backTracking(nums, k, target, i - 1, curSum + nums[i], flag)) return true;
            flag[i] = false;//回溯
            while(i > 0 && nums[i - 1] == nums[i]) i--;//剪枝，重复元素跳过
        }
        return false;
    }
}