面试题40.最小的k个数

题目描述：
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。
解析：时间复杂度为O(nlogn)的算法，特别适用于海量数据处理。先建立一个大小为k的数据容器来存储最小的k个数字，接下来每次从输入的n个整数中读入一个数。如果容器中已有的数字少于k个，则直接把这次读入的整数放入容器之中；如果容器中已有k个数字，即容器已满，此时我们不能在插入数字而实替换让其中的数字。找出这k个数字中的最大值，然后拿这个最大值和待插入的整数进行比较。如果待插入的值小于当前已有的最大值，那么就将这个数字替换已有的当前最大值，如果这个待插入的数字比当前最大数字还要大，那么就将这个数字舍弃，它不可能是最小的k个整数之一。每次都要找到最大数字，因此容易想到用最大堆。最大堆中，根节点的值总是大于它的子树中任意节点的值。
Java代码实现：

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(input == null || k > input.length || k <= 0){
            return list;
        }
        int[] kArray = Arrays.copyOfRange(input, 0, k);
        //创建大顶堆
        buildHeap(kArray);
        for(int i = k; i < input.length; i++){
            if(input[i] < kArray[0]){
                kArray[0] = input[i];
                maxHeap(kArray, 0);
            }
        }
        for(int i = kArray.length-1; i >= 0; i--){
            list.add(kArray[i]);
        }
        return list;
    }
    public void buildHeap(int[] input){
        for(int i = input.length/2 - 1; i >= 0; i--){
            maxHeap(input, i);
        }
    }
    private void maxHeap(int[] array, int i){
        int left = 2*i + 1;
        int right = left + 1;
        int largest = 0;
        if(left < array.length && array[left] > array[i])
            largest = left;
        else{
            largest = i;
        }
        if(right < array.length && array[right] > array[largest])
            largest = right;
        if(largest != i){
            int temp = array[i];
            array[i] = array[largest];
            array[largest] = temp;
            maxHeap(array, largest);
        }
    }
}