1. 程式人生 > >演算法---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);

  穩定性:不穩定

建立大根堆的方法:

//構建大根堆:將array看成完全二叉樹的順序儲存結構
    private int[] buildMaxHeap(int[] array){
        //從最後一個節點array.length-1的父節點(array.length-1-1)/2開始,直到根節點0,反覆調整堆
        for(int i=(array.length-2)/2;i>=0;i--){ 
            adjustDownToUp(array, i,array.length);
        }
        return array;
    }
    
    //將元素array[k]自下往上逐步調整樹形結構
    private void adjustDownToUp(int[] array,int k,int length){
        int temp = array[k];   
        for(int i=2*k+1; i<length-1; i=2*i+1){    //i為初始化為節點k的左孩子,沿節點較大的子節點向下調整
            if(i<length && array[i]<array[i+1]){  //取節點較大的子節點的下標
                i++;   //如果節點的右孩子>左孩子,則取右孩子節點的下標
            }
            if(temp>=array[i]){  //根節點 >=左右子女中關鍵字較大者,調整結束
                break;
            }else{   //根節點 <左右子女中關鍵字較大者
                array[k] = array[i];  //將左右子結點中較大值array[i]調整到雙親節點上
                k = i; //【關鍵】修改k值,以便繼續向下調整
            }
        }
        array[k] = temp;  //被調整的結點的值放人最終位置
    }

建立完成後進行堆排序:

//堆排序
    public int[] heapSort(int[] array){
        array = buildMaxHeap(array); //初始建堆,array[0]為第一趟值最大的元素
        for(int i=array.length-1;i>1;i--){  
            int temp = array[0];  //將堆頂元素和堆低元素交換,即得到當前最大元素正確的排序位置
            array[0] = array[i];
            array[i] = temp;
            adjustDownToUp(array, 0,i);  //整理,將剩餘的元素整理成堆
        }
        return array;
    }

 刪除堆頂元素:

//刪除堆頂元素操作
    public int[] deleteMax(int[] array){
        //將堆的最後一個元素與堆頂元素交換,堆底元素值設為-99999
        array[0] = array[array.length-1];
        array[array.length-1] = -99999;
        //對此時的根節點進行向下調整
        adjustDownToUp(array, 0, array.length);
        return array;
    }

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

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

//插入操作:向大根堆array中插入資料data
    public int[] insertData(int[] array, int data){
        array[array.length-1] = data; //將新節點放在堆的末端
        int k = array.length-1;  //需要調整的節點
        int parent = (k-1)/2;    //雙親節點
        while(parent >=0 && data>array[parent]){
            array[k] = array[parent];  //雙親節點下調
            k = parent;
            if(parent != 0){
                parent = (parent-1)/2;  //繼續向上比較
            }else{  //根節點已調整完畢,跳出迴圈
                break;
            }
        }
        array[k] = data;  //將插入的結點放到正確的位置
        return array;
    }

進行測試:

public void toString(int[] array){
        for(int i:array){
            System.out.print(i+" ");
        }
    }
    
    public static void main(String args[]){
        HeapSort hs = new HeapSort();
        int[] array = {87,45,78,32,17,65,53,9,122};
        System.out.print("構建大根堆:");
        hs.toString(hs.buildMaxHeap(array));
        System.out.print("\n"+"刪除堆頂元素:");
        hs.toString(hs.deleteMax(array));
        System.out.print("\n"+"插入元素63:");
        hs.toString(hs.insertData(array, 63));
        System.out.print("\n"+"大根堆排序:");
        hs.toString(hs.heapSort(array));    
    }
 構建大根堆:122 87 78 45 17 65 53 9 32 
 刪除堆頂元素:87 45 78 32 17 65 53 9 -99999 
 插入元素63:87 63 78 45 17 65 53 9 32 
 大根堆排序:9 17 32 45 53 63 65 78 87