leetcode698. 划分为k个相等的子集

最新推荐文章于 2024-08-25 11:30:06 发布

kkkkuuga

最新推荐文章于 2024-08-25 11:30:06 发布

阅读量642

点赞数

CC 4.0 BY-SA版权

分类专栏：回溯算法文章标签： leetcode 算法数据结构 java

本文链接：https://blog.youkuaiyun.com/kkkkuuga/article/details/126262040

回溯算法专栏收录该内容

18 篇文章

订阅专栏

本文探讨了LeetCode上一道经典题目，即如何判断一个整数数组能否被划分为k个非空子集且每个子集的总和相等。通过回溯算法实现，并运用排序和剪枝策略来优化搜索过程。

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

1.题目描述：

给定一个整数数组nums和一个正整数k，找出是否有可能把这个数组分成 k个非空子集，其总和都相等。

2.回溯：思路见leetcode题解。

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);//剪枝，贪心思路，排序后倒序遍历
        if (nums[nums.length - 1] > target) return false;
        boolean[] flag = new boolean[nums.length];
        return backTracking(nums, k, target, nums.length - 1, 0, flag);
    }

    public boolean backTracking(int[] nums, int k, int target, int index, int curSum, boolean[] flag) {
        if (k == 1) return true;//剪枝，k-1个组合满足条件即可
        if (curSum == target) return backTracking(nums, k - 1, target, nums.length - 1, 0, flag);//找到其中一个组合，再重新找
        for (int i = index; i >= 0; i--) {
            if (flag[i] || curSum + nums[i] > target) continue;//剪枝
            flag[i] = true;
            if (backTracking(nums, k, target, i - 1, curSum + nums[i], flag)) return true;
            flag[i] = false;//回溯
            while(i > 0 && nums[i - 1] == nums[i]) i--;//剪枝，重复元素跳过
        }
        return false;
    }
}