【leetcode】698. Partition to K Equal Sum Subsets

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ghscarecrow/article/details/86559725

Given an array of integers nums and a positive integer k, find whether it's possible to divide this array into k non-empty subsets whose sums are all equal.

Example 1:

Input: nums = [4, 3, 2, 3, 5, 2, 1], k = 4
Output: True
Explanation: It's possible to divide it into 4 subsets (5), (1, 4), (2,3), (2,3) with equal sums.

Note:

  • 1 <= k <= len(nums) <= 16.
  • 0 < nums[i] < 10000.
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);
        int row = nums.length - 1;
        /**
        经过排序,最后的元素为该数组的最大数,如果该数大于目标值,则代表该数组无论如何切割都无法达成目标
        **/
        if(nums[row] > target){
            return false;
        }
        while(row>=0 && nums[row] == target) {//为了找到一个小于目标值的数的下标
            row--;
            k--;
        }
        return search(new int[k],row,nums,target);//开始进行搜索
        
    }
    
    public boolean search(int[] groups, int row, int[] nums, int target){
        if(row<0)
            return true;
        int v = nums[row--];
        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[i] -= v;//复位
            }
            if(groups[i] == 0){
                break;
            }
        }
        return false;
    }
}

猜你喜欢

转载自blog.csdn.net/ghscarecrow/article/details/86559725