题目描述:
输入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);
}
}