给定一个整数数组 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的值)
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; } }
本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。
我来说两句