LeetCoe 698 (Partition to K Equal Sum Subsets)

1.題目描述：
輸入， 一個數組，一個整數 k, 判斷是否可以將 這個數組劃分爲 k 個相等的子數組。
2.例子：
輸入: nums = [4, 3, 2, 3, 5, 2, 1], k = 4
輸出: True
解釋：將 nums劃分爲[5], [1,4], [2,3], [2,3]4個子數組，並且和都爲5。因此返回 true。

3.代碼

class Solution {
    public boolean canPartitionKSubsets(int[] nums, int k) {
        int sum = 0;
        int eachSum = 0;
        for(int num : nums) sum += num;
        if(sum % k != 0) return false;
        eachSum = sum/k;
        int [] boxes = new int[k];
        Arrays.sort(nums);
    //從大到小搜索，保證數值大的每回都插入boxes中可以減少很多搜索次數

        return recursive(nums, k, eachSum, index-1, boxes);
    }

    public boolean recursive(int [] nums, int k, int eachSum, int index, int [] boxes){
        if(k == 0) return true;
        for(int i = 0; i < boxes.length; i++){
            if(boxes[i] + nums[index] > eachSum){
                continue;
            }else if(boxes[i] + nums[index] < eachSum){
                boxes[i] += nums[index];
                if(recursive(nums, k, eachSum, index-1, boxes)) return true;
                boxes[i] -= nums[index];
            }else{
                boxes[i] += nums[index];
                if(recursive(nums, k-1, eachSum, index-1, boxes)) return true;
                boxes[i] -= nums[index];
            }
        }
        return false;
    }
}