剑指Offer第二十九题:最小的k个数

剑指Offer第二十九题:最小的k个数

在这里插入图片描述
使用一个最大堆:根结点最大

  • 最大堆先存储数组中前k个元素,将数组中其余的元素逐一与最大堆顶元素比较,如果小于堆顶元素,就将堆顶元素移除,并添加当前元素。

  • 实现:

  • 使用PriorityQueue来实现最大堆,内置比较器(Integer o1,Integer o2){当o1<o2时,返回-1,否则返回1,可以实现队头元素大于队尾元素,PriorityQueue的底层是用最大堆或者最小堆实现的}

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        if(input == null || k == 0 || k>input.length){
            return list;
        }
        Queue <Integer> queue = new PriorityQueue<>(new Comparator<Integer>(){
			@Override
			public int compare(Integer o1, Integer o2) {
				// TODO Auto-generated method stub
				if(o1<o2) {
					return 1;
				}else {
					return -1;
				}
			}
        });
        for(int i = 0; i<k; i++){
        	queue.add(input[i]);
        }
        for(int i = k; i<input.length; i++){
            if(queue.peek()>input[i]){
            	queue.poll();
            	queue.offer(input[i]);
            }
        }
        while(!queue.isEmpty()){
            list.add(queue.poll());
        }
        return list;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值