剑指offer——最小的K个数

最小的K个数

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

排序——堆排序

利用堆排序中的构造小顶堆的方法，依次构造k次，然后就可以求出最小的k个数

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res = new ArrayList<>();
        if(k > input.length) return res;
        int len = input.length;
        for(int i = 0; i < k; i++){
            smallHeap(input, len--);
            int temp = input[0];
            input[0] = input[len];
            input[len] = temp;
            res.add(input[len]);
        }
        return res;
    }
    public void smallHeap(int[] input, int high){
        for(int i = high/2; i > 0; i--){
            int min;
            if( 2 * i >= high) min = 2 * i;
            else{
                min = input[2 * i - 1] < input[2 * i] ? 2 * i : 2 * i + 1;
            }
            if(input[i - 1] > input[min - 1]) {
                int temp = input[i - 1];
                input[i - 1] = input[min - 1];
                input[min - 1] = temp;
            }
        }
    }
}