Leetcode 每日一题 2022-09-20 698.划分k个相等的子集

最新推荐文章于 2024-07-14 16:07:02 发布

same-pxt

最新推荐文章于 2024-07-14 16:07:02 发布

阅读量53

点赞数

分类专栏： leetcoed每日一题监督学习文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_61509617/article/details/126953973

版权

leetcoed每日一题监督学习专栏收录该内容

17 篇文章 0 订阅

订阅专栏

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

在这里插入图片描述

对于本题，首先去排除不可能的情况。
如果数组的和不能整除k，则不满足要求。
如果有某个元素大于sum/k，则不满足要求。

排除完特殊情况就该进入正题，创建一个bucket数组记录每个子数组的和。

再整个过程中需要给bucket[i]加上num[m]，如果不合适去掉，所以可以想到回溯算法。

选择参数index，如果index==nums.size 那么就结束。
如果bucket[i]+num[m]>sum/k,则不符合要求，然后退出num[m]再进一步回溯。

class Solution {
public:
    bool canPartitionKSubsets(vector<int>& nums, int k) {
        if(k > nums.size()) return false;
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if(sum % k != 0) return false;
        sort(nums.rbegin(), nums.rend());
        bucket.resize(k); 
        int target = sum/k;
        return backtrack(nums, 0, target);
    }

private:
    // 记录每个子集中数字之和
    vector<int> bucket;

    bool backtrack(vector<int> &nums, int index, int target){
        if(index == nums.size()){
            return true;
        }

        for(int i = 0; i < bucket.size(); i++){
            if(bucket[i] + nums[index] > target){
                continue;
            }
            if(i > 0 && bucket[i] == bucket[i-1]){
                continue;
            }
            bucket[i] += nums[index];
            if(backtrack(nums, index + 1, target)){
                return true;
            }
            bucket[i] -= nums[index]; 
        }
        return false;
    }
};