java topK问题 最大堆实现

题目描述：

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

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
          ArrayList<Integer>topk=new ArrayList();
        if(k>input.length){
        return topk;
        }
        buildMaxHeap(input,k);
        for(int i=k;i<input.length;i++){
            if(input[i]<input[0]){//i对应的值小于所构建的最大堆的最大值，则需要对最大堆进行重新构建，即用input[i]替换掉最大堆的最大值，然后构建最大堆。
                input[0]=input[i];
                heapify(input,0,k);
            }
        }
        for(int i=0;i<k&&i<input.length;i++)
            topk.add(input[i]);
        return topk;
    }
    
     //Second Step: 考虑到二叉树的性质，按照从k/2->0的顺序构建最大堆，而无需从k-1处（最后一个叶子节点的所在位置）开始构建
    private void buildMaxHeap(int []input ,int k){
        for(int i=k/2;i>=0;i--)
            heapify(input,i,k);
    }
    
    //First  Step: 以index所在节点作为根节点，对子树构建最大堆
    private void heapify(int[]input,int index,int k){
        int left=((index+1)<<1)-1;   //左子树索引：（考虑到索引是从0开始，而非从1开始）
        int right=left+1;//右子树索引
        int largest=index;//将此子树的最大值的索引设为index
        if(left<k&&input[left]>input[largest])//左子树的值大于largest对应的值
            largest=left;
        if(right<k&&input[right]>input[largest])//右子树的值大于largest对应的值
            largest=right;
        if(largest==index)return;//index没发生改变，说明index对应的值大于左右子树的值
        //否则，交换，并对子树的子树构建最大堆
        int temp = input[index];
        input[index]=input[largest];
        input[largest]=temp;
        heapify(input,largest,k);
        
    }
}