划分为K个相同的子集(递归)

18 篇文章 0 订阅
10 篇文章 0 订阅

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

示例 1:

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

初始判断:由题目我们可知数组 nums 的总和为 sums,必须满足 sums%k = 0,且nums中的任何数都不能大于 target = sums/k,这是初始的检验条件。

总体思想:我们新建一个数组 int []groups = new int[k]; 用来存放 k 个集合。而我们递归的步骤则是:每一步从 nums 数组 中取出一个数字将其放入 groups 数组中,随后进行判断放入后的 groups[i] 是否大于 target
(1)如果不大于,则说明还可以继续存放 nums数组中的数,递归即可。
(2)如果大于,则说明不符,groups[i] 减去存放的那个数,i++。

终止条件:当nums 数组中的数全部遍历过后说明已经找到了一种符合题意的答案,return true即可。

class Solution {
    public boolean canPartitionKSubsets(int[] nums, int k) {
        int sum= 0;
        int max = 0;
        for(int i = 0;i<nums.length;i++) {
        	if(nums[i]>max) {
        		max = nums[i];
        	}
        	sum +=nums[i];
        }
        if(sum%k!=0||max>sum/k) {//必须满足 **sums%k = 0**,且nums中的任何数
        //都不能大于 **target = sums/k**
        	return false;
        }
        int target = sum/k;
        Arrays.sort(nums);
        int row = nums.length-1;
        while(row>=0&&nums[row]==target) {//从Array.sort(nums)开始到此处,先排序,是为了将
        //等于target的值去掉,优化时间,前面判断最大值是否大于target
        //其实可以用nums[nums.length-1]判断,为了简明,就不做变换
        //(如果不明白,可略过,不影响过程)
        	row--;
        	k--;
        }
        return helper(new int[k],nums,row,target);
    }
	public boolean helper(int []a,int []nums,int row,int target) {
		 if(row<0) {//截至条件
			 return true;
		 }
		 for(int i = 0;i<a.length;i++) {
			 if(a[i]+nums[row]<=target) {
				 a[i] += nums[row];
				 if(helper(a,nums,row-1,target)) {//如果不超过target便继续递归
					 return true;
				 }
				 a[i] -= nums[row];//可以运行到此处说明上一种解法并没有成功,所以减去加上的值
			 }
			 if(a[i] == 0) {//此处是为了剪枝当一个数放在a[i] = 0处也不成功,
			 //放在其他为0处也必然不成立
				 break;
			 }
		 }
		 return false;
	}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值