算法:划分数组为连续数字的集合

给定一个整数数组nums和一个正整数k,请判断,正整数k是否能将数组nums再划分成若干个长度为k的连续数字的数组?

例:

输入:nums = [1,2,3,3,4,4,5,6], k = 4
输出:true
解释:数组可以分成 [1,2,3,4] 和 [3,4,5,6]。

题解:

这里我使用了优先队列PriorityQueue,优先队列PriorityQueue.poll()能保证取出的元素永远是权值最小的。

代码如下:

class Solution {
    public boolean isPossibleDivide(int[] nums, int k) {
        int len = nums.length;
        // 数组长度一定要是k的倍数才可以划分
        if(len % k != 0){
            return false;
        }

        // 优先队列的作用是能保证每次取出的元素都是队列中权值最小的
        PriorityQueue<Integer> heap = new PriorityQueue<>(len);
        for(int num : nums){
            heap.offer(num);
        }

        while(!heap.isEmpty()){
            // top:队列中权值最小的元素
            int top = heap.poll();

            // 除了top,再取k-1个元素
            for(int i =1; i < k ; i++){
                // i是相对于top的偏移量
                if(!heap.remove(top + i)){
                    return false;
                }
            }
        }
        return true;
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值