題目描述

輸入n個整數，找出其中最小的K個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4,。

思路

1. 此問題屬于TopK問題，題目中要求取最小的數，所以是建立大根堆。

2.可以借助java原生的優(yōu)先級隊列，底層就是按照建堆思想實現的。

Java代碼實現

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if(k>input.length || k==0){
            return new ArrayList<>();
        }
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k,(o1,o2)->(o2-o1));

        for (int i = 0; i <input.length; i++) {
            if(priorityQueue.size()<k){
                priorityQueue.add(input[i]);
            }else{
                if(priorityQueue.peek()>input[i]){
                    priorityQueue.poll();
                    priorityQueue.add(input[i]);
                }
            }
        }
        return new ArrayList<>(priorityQueue);
    }

}