最小的k个数 (java)

题目描述:

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

解题思路:

使用优先队列保证队列里存放的是最小的K位数

第一次在解题过程中使用到优先队列,做一下记录

import java.util.ArrayList;
import java.util.Comparator;
import java.util.PriorityQueue;
public class Solution {
    //注意优先队列在使用时再声明,注意其声明方法
    public ArrayList GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList arrayList = new ArrayList();
        if(k > input.length || k <= 0){
            return arrayList;
        }
        PriorityQueue minQueue = new PriorityQueue(k, new Comparator(){
            
            @Override
            public int compare(Integer o1, Integer o2){
                return o2.compareTo(o1);
            }
        });
        for(int j = 0; j < input.length; j++){
            if(j < k){
                minQueue.offer(input[j]);
            }else{
                if(minQueue.peek() > input[j]){
                    minQueue.poll();
                    minQueue.offer(input[j]);
                }
            }
        }
        arrayList.addAll(minQueue);
        return arrayList;
    }
}

 

你可能感兴趣的:(算法+数据结构)