zoukankan      html  css  js  c++  java
  • 698.划分为k个相等的子集

    回溯

    预处理:
    1.计算累加和s,如果s%k!=0,返回false。
    2.按从小到大给数组排序。
    准备k个桶,每个桶的容量为s/k。遍历数组,尝试将当前元素添加到每个桶中,如果所有所有桶都放不下则返回false。如果能遍历完数组则返回true。
    数组排序是一种优化操作,如果当前元素都没有桶可以放得下,那后面的元素就不用再遍历可以直接返回false了,因为后面的元素更大。

       public boolean canPartitionKSubsets(int[] nums, int k) {
            //因为题目限制条件不用担心溢出
            int sum = 0;
            for(int i = 0; i < nums.length; i++){
                sum += nums[i];
            }
            if(sum % k != 0){
                return false;
            }
            //求出子集的和
            sum = sum / k;
            //排序 小的放最前面大的放最后面
            Arrays.sort(nums);
            //如果子集的和小于数组最大的直接返回false
            if(nums[nums.length - 1] > sum){
                return false;
            }
            //建立一个长度为k的桶
            int[] arr = new int[k];
            //桶的每一个值都是子集的和
            Arrays.fill(arr, sum);
            //从数组最后一个数开始进行递归
            return help(nums, nums.length - 1, arr, k);
        }
        
        boolean help(int[] nums, int cur, int[] arr, int k){
            //已经遍历到了-1说明前面的所有数都正好可以放入桶里,那所有桶的值此时都为0,说明找到了结果,返回true
            if(cur < 0){
                return true;
            }
            //遍历k个桶
            for(int i = 0; i < k; i++){
                //如果正好能放下当前的数或者放下当前的数后,还有机会继续放前面的数(剪枝)
                if(arr[i] == nums[cur] || (cur > 0 && arr[i] - nums[cur] >= nums[0])){
                    //放当前的数到桶i里
                    arr[i] -= nums[cur];
                    //开始放下一个数
                    if(help(nums, cur - 1, arr, k)){
                        return true;
                    }
                    //这个数不该放在桶i中
                    //从桶中拿回当前的数
                    arr[i] += nums[cur];
                }
            }
            return false;
        }
    
  • 相关阅读:
    登录 要求 密码必须包含数字、大写字母、小写字母、特殊字字符4种中至少3种
    job 异步获取
    g++的使用
    extern "c"
    C/C++定义全局变量/常量几种方法的区别
    使用Github管理cocos2dx版本
    二维码限制扫描区域
    Xcode路径改变之后,Cocopods不能使用,一行命令解决
    iOS之精确定时器
    iOS开发/App安全/代码自动混淆笔记
  • 原文地址:https://www.cnblogs.com/Frank-Hong/p/15472620.html
Copyright © 2011-2022 走看看