前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Leetcode 698. 划分为k个相等的子集

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

作者头像
glm233
发布2020-09-28 17:44:59
1.1K0
发布2020-09-28 17:44:59
举报

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

示例 1:

输入: nums = [4, 3, 2, 3, 5, 2, 1], k = 4 输出: True 说明: 有可能将其分成 4 个子集(5),(1,4),(2,3),(2,3)等于总和。

注意:

1 <= k <= len(nums) <= 16 0 < nums[i] < 10000

算是比较慢的解法吧,先判断数组总和是否是k的倍数,不是就false,然后置一vis数组标记是否访问过,dfs从0到k-1 共k个集合,做完一个做下一个(做完的意思是填正好sum/k的值)

代码语言:javascript
复制
class Solution {
    public boolean canPartitionKSubsets(int[] nums, int k) {
        int sum = 0;
        int len = nums.length;
        for (int i = 0; i < len; i++) 
            sum += nums[i];
        
        if(sum % k != 0 ) return false;
        int avg = sum / k;
        boolean[] flag = new boolean[len];
        return help(nums,flag,avg,k,avg,0);
    }
    
    public static boolean help(int[] nums, boolean[] flag, int avg, int k, int temp, int index ){
 
        if (k == 0 ) return true;
 
        if (temp == 0)
            return help(nums,flag,avg,k-1,avg,0);
 
        for (int i = index; i < nums.length; i++) {
            if (flag[i] == true) continue;
 
            flag[i] = true;
            if(temp-nums[i]>=0 && help(nums,flag,avg,k,temp-nums[i],index+1)){
                return true;
            }
            flag[i] = false;
 
        }
 
        return false;
    }
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-03-11 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档