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

最新推荐文章于 2024-10-06 14:42:09 发布

咔咔咔的

最新推荐文章于 2024-10-06 14:42:09 发布

阅读量141

点赞数 3

分类专栏： leetcode 文章标签： c++

本文链接：https://blog.csdn.net/weixin_45256307/article/details/141803547

版权

leetcode 专栏收录该内容

591 篇文章 2 订阅

订阅专栏

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

题目链接：698. 划分为k个相等的子集

代码如下：

class Solution 
{
public:
    bool canPartitionKSubsets(vector<int>& nums, int k) 
    {
        int sum=accumulate(nums.begin(),nums.end(),0);
        if(sum%k>0)   {return false;}
        int per=sum/k;
        vector<int> cur(k,0);

        function<bool(int)> dfs=[&](int i)->bool
        {
            if(i==nums.size())    {return true;}

            for(int j=0;j<k;j++)
            {
                if(j&&cur[j]==cur[j-1]) {continue;}
                cur[j]+=nums[i];
                if(cur[j]<=per&&dfs(i+1)) {return true;}
                cur[j]-=nums[i];
            }
            return false;
        };

        sort(nums.begin(),nums.end(),greater<int>());
        return dfs(0);
    }
};