【leetcode】698. Partition to K Equal Sum Subsets

最新推荐文章于 2025-02-03 15:37:10 发布

原创最新推荐文章于 2025-02-03 15:37:10 发布 · 159 阅读

0 ·

CC 4.0 BY-SA版权

解题报告同时被 2 个专栏收录

203 篇文章

订阅专栏

leetcode题解大全

115 篇文章

订阅专栏

本文探讨了如何判断一个整数数组是否可以被划分为k个非空子集，且每个子集的和相等的问题。通过示例说明，如输入数组[4,3,2,3,5,2,1]与k=4，解释了算法的实现过程，包括求和、排序、搜索等关键步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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