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 GetLeastNumbers_Solution(int [] input, int k) {
          ArrayListtopk=new ArrayList();
        if(k>input.length){
        return topk;
        }
        buildMaxHeap(input,k);
        for(int i=k;i0的顺序构建最大堆，而无需从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(leftinput[largest])//左子树的值大于largest对应的值
            largest=left;
        if(rightinput[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);
        
    }
}