1. 程式人生 > >LeetCoe 698 (Partition to K Equal Sum Subsets)

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