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");

}

}

總結

以上是生活随笔為你收集整理的python堆排序求topn_Java堆排序，取得前TopN个数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。