698. Partition to K Equal Sum Subsets

Medium
L家的tag题,之前contest好像是做过但没有做出来当时,这道题看了答案然后写出了一个及其奇怪的bug.

class Solution {
    public boolean canPartitionKSubsets(int[] nums, int k) {
        int sum = 0;
        for (int num : nums){
            sum += num;
        }
        if (sum % k != 0){
            return false;
        }
        int subSum = sum / k;
        Arrays.sort(nums);
        if (nums[nums.length - 1] > subSum){
            return false;
        }
        int beginIndex = nums.length - 1;
        while (beginIndex >= 0 && nums[beginIndex] == subSum){
            beginIndex--;
            k--;
        }
        return canPartition(new int[k], nums, subSum, beginIndex);
    }

    private boolean canPartition(int[] subsets, int[] nums, int target, int index){
        if (index < 0){
            return true;
        }
        int curtNum = nums[index];
        for (int i = 0; i < subsets.length; i++){
            if (subsets[i] + curtNum <= target){
                subsets[i] += curtNum;
                if (canPartition(subsets, nums, target, index - 1)){
                    return true;
                }
                subsets[i] -= curtNum;
            }
        }
        return false;
    }
}

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容