import java.util.Random;
/**
* Created with IntelliJ IDEA.
* User: pengfei.hpf
* Date: 14-4-29
* Time: 上午11:45
* To change this template use File | Settings | File Templates.
*/
public class HeapSortUtil {
/**
* 用堆排序方法 找出前N个最大的数
* @originalArray 原始数据数组
* @topN 需要取得的N个最大数
* @return 包含topN个最大数的数组
*/
public int[] getTopArray(int[] originalArray,int topN){
int len = originalArray.length ;
if(len <= topN){
return originalArray;
}
int[] array = new int [topN];
initHeap(originalArray);
int temp;
for(int i=0;i
array[i]= originalArray[0];
temp=originalArray[originalArray.length-i-1];
originalArray[originalArray.length-i-1]=originalArray[0];
originalArray[0]=temp;
buildHeap(0,originalArray.length-i-1,originalArray);
}
return array;
}
/**
* 创建初始无序堆
*/
private void initHeap(int[] orignalArr){
for(int i=orignalArr.length-1;i>=0;i--){
buildHeap(i,orignalArr.length,orignalArr);
}
}
/**
* 调整堆
* @param location 起始位置
* @param unSortLength 无序堆的长度
*/
private void buildHeap(int location,int unSortLength,int[] arr){
int temp;
int tempLoc;
//判断该父节点是否有左右孩子
if((tempLoc = (location+1)*2)
if(arr[tempLoc]>arr[tempLoc-1]){//如果右节点大于左节点
if(arr[tempLoc]>arr[location]){//如果右节点大于父节点 就双方交换值
temp = arr[location];
arr[location] = arr[tempLoc];
arr[tempLoc] = temp;
buildHeap(tempLoc,unSortLength,arr);//递归
}
}else{//如果左节点大于右节点
if(arr[tempLoc-1]>arr[location]){//如果左节点大于父节点
temp = arr[location];
arr[location] = arr[tempLoc-1];
arr[tempLoc-1] = temp;
buildHeap(tempLoc-1,unSortLength,arr);//递归
}
}
}else if((tempLoc =((location+1)*2-1))
if(arr[tempLoc]>arr[location]){//如果右节点大于父节点
temp = arr[location];
arr[location] = arr[tempLoc];
arr[tempLoc] = temp;
buildHeap(tempLoc,unSortLength,arr);//递归
}
}
}
public static void main(String[] args) {
int[] arr =new int[100000];
Random ran = new Random();
for (int i = 0; i < arr.length; i++) {
arr[i] = ran.nextInt(100000);
}
HeapSortUtil h = new HeapSortUtil();
long start = System.currentTimeMillis();
int topArr[] = h.getTopArray(arr, 20);
//打印出排序后的数组
for(int i=0;i
System.out.println(topArr[i]);
}
long end = System.currentTimeMillis()-start;
System.out.println("Total time：" + end + "ms");
}
}
 -- Canvas)
