一、题目:
给定一个整数数组 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) <= 160 < nums[i] < 10000
二、思路:
首先是题意:做每道算法题,最重要的是,读懂题意,从剖析咱们能够得进去的是,目标值及每个子集的和target=sum(nums)/k,所以就确定了目标值target。
得出目标值后,咱们能够失去一些肯定为false的状况,如
- 目标值不为整数
- nums中有值大于target的
咱们还能够先找出nums中是否有等于target的值,
这道题的思路就是把nums所有元素都放进长度为k的groups数组当中,如果刚刚放完,则返回true。
对于放进groups这个动作咱们称它为搜寻元素放进group,对于一共k个子集的咱们只须要一个for循环就能解决。
应用递归咱们能够做到始终搜寻。只用扭转以后搜寻的值,就能始终向group数组中增加。
如果看不懂思路,请联合代码剖析。
三、代码:
public class 划分为k个相等的子集 { public static void main(String[] args) { 划分为k个相等的子集 a = new 划分为k个相等的子集(); int[] nums = {4, 3, 2, 3, 5, 2, 1}; System.out.println(a.canPartitionKSubsets(nums,4)); } public boolean canPartitionKSubsets(int[] nums, int k) { int sum = Arrays.stream(nums).sum(); if (sum % k > 0) return false; //由题意咱们能够晓得目标值相对是sum/k int target = sum / k; //给nums数组排序,递增程序 Arrays.sort(nums); int row = nums.length - 1; //如果最大的值大于目标值,则改数组必定不满足条件,间接返回false if (nums[row] > target) return false; //如果有等于目标值的,k--,row-- while (row >= 0 && nums[row] == target) { row--; k--; } return search(new int[k], row, nums, target); } //函数作用:搜寻nums中的每一个数,能不能增加到groups里,并且和等于target,row代表以后增加到groups中的数 public boolean search(int[] groups, int row, int[] nums, int target) { //如果nums中没有数了,则代表所有数全副放入groups中 if (row < 0) return true; //留神row--,先赋值,再-- int v = nums[row--]; //遍历groups的k个子集 for (int i = 0; i < groups.length; i++) { if (groups[i] + v <= target) { groups[i] += v; if (search(groups, row, nums, target)) return true; //不能增加到groups中,则把它减去,及回溯 groups[i] -= v; } //如果遍历完了,所有数都不能放进groups中,及没有值相加或者自身等于target,则跳出循环break,或者return false; if (groups[i] == 0) return false; } return false; }}