JZ29 最小的K个数

原题链接

描述

---

给定一个数组，找出其中最小的K个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

0 <= k <= input.length <= 10000；

0 <= input[i] <= 10000。

示例1

输入：[4,5,1,6,2,7,3,8],4 
返回值：[1,2,3,4]
说明：返回最小的4个数即可，返回[1,3,2,4]也可以 

示例2

输入：[1], 0
返回值：[]

示例3

输入：[0,1,2,1,2],3
返回值：[0,1,1]

思路

一、最简单的办法，把input[]元素全部插入PriorityQueue中进行自动排序，然后k次poll()出队首元素，这样时间复杂度是。

二、用堆的方法

三、分治。

解答

一、简单

import java.util.ArrayList;
import java.util.PriorityQueue;

public class Solution {
    public static ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if(input.length==0)
            return new ArrayList<>(0);
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(input.length);
        ArrayList<Integer> arrayList = new ArrayList<>(4);
        for (int i=0;i<input.length;i++){
            priorityQueue.add(input[i]);
        }

        for (int i=0;i<k;i++){
            arrayList.add(priorityQueue.poll());
        }

        return arrayList;
    }

    public static void main(String[] args) {
        int[] arr = {4,5,1,6,2,7,3,8};

        System.out.println(GetLeastNumbers_Solution(arr, 4));
    }
}

二、堆

还没写。

三、分治

没写。