java 實現大頂堆


Java實現堆排序(大根堆)

 

  堆排序是一種樹形選擇排序方法,它的特點是:在排序的過程中,將array[0,...,n-1]看成是一顆完全二叉樹的順序存儲結構,利用完全二叉樹中雙親節點和孩子結點之間的內在關系,在當前無序區中選擇關鍵字最大(最小)的元素。

1. 若array[0,...,n-1]表示一顆完全二叉樹的順序存儲模式,則雙親節點指針和孩子結點指針之間的內在關系如下:

  任意一節點指針 i:父節點:i==0 ? null : (i-1)/2

            左孩子:2*i + 1

            右孩子:2*i + 2

2. 堆的定義:n個關鍵字序列array[0,...,n-1],當且僅當滿足下列要求:(0 <= i <= (n-1)/2)

      ① array[i] <= array[2*i + 1] 且 array[i] <= array[2*i + 2]; 稱為小根堆;

      ② array[i] >= array[2*i + 1] 且 array[i] >= array[2*i + 2]; 稱為大根堆;

3. 建立大根堆:

  n個節點的完全二叉樹array[0,...,n-1],最后一個節點n-1是第(n-1-1)/2個節點的孩子。對第(n-1-1)/2個節點為根的子樹調整,使該子樹稱為堆。

  對於大根堆,調整方法為:若【根節點的關鍵字】小於【左右子女中關鍵字較大者】,則交換。

  之后向前依次對各節點((n-2)/2 - 1)~ 0為根的子樹進行調整,看該節點值是否大於其左右子節點的值,若不是,將左右子節點中較大值與之交換,交換后可能會破壞下一級堆,於是繼續采用上述方法構建下一級的堆,直到以該節點為根的子樹構成堆為止。

  反復利用上述調整堆的方法建堆,直到根節點。

4.堆排序:(大根堆)

  ①將存放在array[0,...,n-1]中的n個元素建成初始堆;

  ②將堆頂元素與堆底元素進行交換,則序列的最大值即已放到正確的位置;

  ③但此時堆被破壞,將堆頂元素向下調整使其繼續保持大根堆的性質,再重復第②③步,直到堆中僅剩下一個元素為止。

堆排序算法的性能分析:

  空間復雜度:o(1);

  時間復雜度:建堆:o(n),每次調整o(log n),故最好、最壞、平均情況下:o(n*logn);

  穩定性:不穩定

 

建立大根堆的方法:

復制代碼
 1     //構建大根堆:將array看成完全二叉樹的順序存儲結構
 2     private int[] buildMaxHeap(int[] array){
 3         //從最后一個節點array.length-1的父節點(array.length-1-1)/2開始,直到根節點0,反復調整堆
 4         for(int i=(array.length-2)/2;i>=0;i--){ 
 5             adjustDownToUp(array, i,array.length);
 6         }
 7         return array;
 8     }
 9     
10     //將元素array[k]自下往上逐步調整樹形結構
11     private void adjustDownToUp(int[] array,int k,int length){
12         int temp = array[k];   
13         for(int i=2*k+1; i<length-1; i=2*i+1){    //i為初始化為節點k的左孩子,沿節點較大的子節點向下調整
14             if(i<length && array[i]<array[i+1]){  //取節點較大的子節點的下標
15                 i++;   //如果節點的右孩子>左孩子,則取右孩子節點的下標
16             }
17             if(temp>=array[i]){  //根節點 >=左右子女中關鍵字較大者,調整結束
18                 break;
19             }else{   //根節點 <左右子女中關鍵字較大者
20                 array[k] = array[i];  //將左右子結點中較大值array[i]調整到雙親節點上
21                 k = i; //【關鍵】修改k值,以便繼續向下調整
22             }
23         }
24         array[k] = temp;  //被調整的結點的值放人最終位置
25     }    
復制代碼

堆排序:

復制代碼
 1     //堆排序
 2     public int[] heapSort(int[] array){
 3         array = buildMaxHeap(array); //初始建堆,array[0]為第一趟值最大的元素
 4         for(int i=array.length-1;i>1;i--){  
 5             int temp = array[0];  //將堆頂元素和堆低元素交換,即得到當前最大元素正確的排序位置
 6             array[0] = array[i];
 7             array[i] = temp;
 8             adjustDownToUp(array, 0,i);  //整理,將剩余的元素整理成堆
 9         }
10         return array;
11     }
復制代碼

刪除堆頂元素(即序列中的最大值):先將堆的最后一個元素與堆頂元素交換,由於此時堆的性質被破壞,需對此時的根節點進行向下調整操作。

復制代碼
1     //刪除堆頂元素操作
2     public int[] deleteMax(int[] array){
3         //將堆的最后一個元素與堆頂元素交換,堆底元素值設為-99999
4         array[0] = array[array.length-1];
5         array[array.length-1] = -99999;
6         //對此時的根節點進行向下調整
7         adjustDownToUp(array, 0, array.length);
8         return array;
9     }
復制代碼

對堆的插入操作:先將新節點放在堆的末端,再對這個新節點執行向上調整操作。

假設數組的最后一個元素array[array.length-1]為空,新插入的結點初始時放置在此處。

復制代碼
 1     //插入操作:向大根堆array中插入數據data
 2     public int[] insertData(int[] array, int data){
 3         array[array.length-1] = data; //將新節點放在堆的末端
 4         int k = array.length-1;  //需要調整的節點
 5         int parent = (k-1)/2;    //雙親節點
 6         while(parent >=0 && data>array[parent]){
 7             array[k] = array[parent];  //雙親節點下調
 8             k = parent;
 9             if(parent != 0){
10                 parent = (parent-1)/2;  //繼續向上比較
11             }else{  //根節點已調整完畢,跳出循環
12                 break;
13             }
14         }
15         array[k] = data;  //將插入的結點放到正確的位置
16         return array;
17     }
復制代碼

 

測試:

復制代碼
 1     public void toString(int[] array){
 2         for(int i:array){
 3             System.out.print(i+" ");
 4         }
 5     }
 6     
 7     public static void main(String args[]){
 8         HeapSort hs = new HeapSort();
 9         int[] array = {87,45,78,32,17,65,53,9,122};
10         System.out.print("構建大根堆:");
11         hs.toString(hs.buildMaxHeap(array));
12         System.out.print("\n"+"刪除堆頂元素:");
13         hs.toString(hs.deleteMax(array));
14         System.out.print("\n"+"插入元素63:");
15         hs.toString(hs.insertData(array, 63));
16         System.out.print("\n"+"大根堆排序:");
17         hs.toString(hs.heapSort(array));    
18     }
復制代碼


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM