【算法】使用大根堆，解决top_k问题

题目描述

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

public static ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list=new ArrayList<Integer>();
if(input==null || input.length<k||input.length<=0 ){
return list;
}
for(int len=(input.length-1)/2; len>=0; len--){
adjustBigHeap(input,len,input.length-1);//初始化建大根堆
}

int tmp;
for(int i=input.length-1; i>0; i--){
//第一个元素交换到最后，重新调整为大根堆
tmp=input[0];
input[0]=input[i];
input[i]=tmp;
adjustBigHeap(input,0,i-1);

}
for(int j=0; j<k; j++){
list.add(input[j]);
}
return list;
}

public static void adjustBigHeap(int[] input, int begin, int length){
int temp;
int child;
for(temp=input[begin]; 2*begin+1<=length; begin=child){//从begin元素向下查找
child=2*begin+1;
if(child<length && input[child]<input[child+1]){//找到孩子中大的
child++;
}
if(input[child]>temp){
input[begin]=input[child];
}else{
break;
}
}
input[begin]=temp;
}
public static void main(String[] args) {
int arr[]={2,3,5,6,1,4,7,8,9};
ArrayList<Integer>a= GetLeastNumbers_Solution(arr,4);
System.out.println(a);
}