剑指offer：40.最小的K个数

题目：给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。
数据范围：0≤k,n≤10000，数组中每个数的大小0≤val≤1000
要求：空间复杂度 O(n) ，时间复杂度 O(nlogn)

示例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]

方法一：使用堆排序

import java.util.ArrayList;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        if(input.length == 0 || k == 0 || k > input.length)
            return list;
        
        //创建小顶堆
        for(int i = input.length / 2 - 1; i >= 0; i--) {
            adjustHeap(input, i, input.length);
        }
        
        //排序
        for(int j = input.length - 1; j >= 0; j--) {
            int first = input[0];
            list.add(first);
            if(list.size() == k)
                return list;
            input[0] = input[j];
            adjustHeap(input, 0, j);
        }
        return list;
    }
    
    private void adjustHeap(int[] input, int i, int length) {
        int temp = input[i];
        for(int l = 2 * i + 1; l < length; l = 2 * l + 1) {
            if(l + 1 < length && input[l] > input[l + 1]) {
                l++;
            } 
            if(input[l] < temp){
                input[i] = input[l];
                i = l;
            } else {
                break;
            }
        }
        input[i] = temp;
    }
}

方法二：使用归并排序

import java.util.ArrayList;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        if(input.length == 0 || k == 0 || k > input.length)
            return list;
        int[] temp = new int[input.length];
        input = mergeSort(input, 0, input.length - 1, temp);
        for(int i = 0; i < k; i ++) {
            list.add(input[i]);
        }
        return list;
    }
    
    private int[] mergeSort(int[] input, int left, int right, int[] temp) {
        if(left < right) {
            int mid = (left + right) / 2;
            mergeSort(input, left, mid, temp);
            mergeSort(input, mid + 1, right, temp);
            merge(input, left, mid, right, temp);
        }
        return input;
    }
    
    //合并
    private void merge(int[] input, int left, int mid, int right, int[] temp) {
        int i = left;
        int j = mid + 1;
        int t = 0;
        
        while(i <= mid && j <= right) {
            if(input[i] < input[j]) {
                temp[t] = input[i];
                t++;
                i++;
            } else {
                temp[t] = input[j];
                t++;
                j++;
            }
        }
        //若左边有剩余
        while(i <= mid) {
            temp[t] = input[i];
            i++;
            t++;
        }
        //若右边有剩余
        while(j <= right) {
            temp[t] = input[j];
            j++;
            t++;
        }
        //将temp数组的元素拷贝到input
        t = 0;
        int tempLeft = left;
        while(tempLeft <= right) {
            input[tempLeft] = temp[t];
            t++;
            tempLeft++;
        }
    }
}