[Java]输入n个整数,找出其中最小的K个数。(两种方法:排序和借助优先队列(最大堆))

题目链接:https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.PriorityQueue;
/*
* 输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
* */
public class GetLeastNumbers_Solution {

    public ArrayList<Integer> getLeastNumbers_Solution(int[] intput,int k ){//排序

        ArrayList<Integer> result=new ArrayList<>();
        if(intput.length==0){
            return result;
        }
        if(k>intput.length){
            return result;
        }

        Arrays.sort(intput);
        for(int i=0;i<k;i++){
            result.add(intput[i]);
        }
        return result;
    }

    public ArrayList<Integer> getLeastNumbers_Solution1(int [] input,int k){//借助优先队列
        ArrayList<Integer> result=new ArrayList<>();
        if(input==null||input.length<k||k<=0){
            return result;
        }
        PriorityQueue<Integer> queue=new PriorityQueue<>(Comparator.reverseOrder());//使用优先队列来实现最大堆。需要设置比较器为逆序
        for(int i=0;i<input.length;i++){
            if(i<k){
                queue.add(input[i]);//最大堆会自动进行排序
            }else{
                if(input[i]<queue.peek()){//如果数组值小于堆顶值
                    queue.poll();//删除堆顶值
                    queue.add(input[i]);//加入数组值然后进行重新排序
                }
            }

        }
        for(int i=0;i<k;i++){
            result.add(queue.poll());//保存堆中元素到结果集中
        }
        return result;

    }


}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值