二叉堆

二叉堆是優(yōu)先隊列很普遍的一種實現(xiàn),它又分為最小堆最大堆,最小堆和最大堆都是完全二叉樹。
其結(jié)構(gòu)體定義如下:

struct HeapStruct {
    int Capacity;    //堆的最大容量
    int Size;         //堆的當前結(jié)點數(shù)
    ElementType *Elements;   //存放堆結(jié)點的數(shù)組
}

typedef struct HeapStruct *PriorityQueue;

二叉堆的結(jié)點可以保存在一個數(shù)組中。
1、如果從數(shù)組的索引1開始存放堆結(jié)點,那么對于數(shù)組中任意位置i上的元素,其左兒子在位置2i上,右兒子在左兒子后的單元(2i + 1)中
2、如果從數(shù)組的索引0開始存放堆結(jié)點,那么對于數(shù)組中任意位置i上的元素,其左兒子在位置(2i+1)上,右兒子在左兒子后的單元(2i+2)中。

在下面的代碼中,統(tǒng)一采用第一種方式存放堆結(jié)點。你可以通過下面的代碼注意到,用第一種方式給編程帶來的好處。

最小堆:父結(jié)點的鍵值總是小于或等于任何一個子節(jié)點的鍵值。

(1)插入操作Insert,采用制造“空穴”,上濾的方式——“空穴”在堆中不斷上移直到找出正確的位置。插入操作復雜度O(logN)。

void Insert(ElementType X, PriorityQueue H) {
    int i;

    if( IsFull(H) ) {      //判斷堆是否已滿
        Error( "Priority queue is full" );
        return ;
    }

    for( i = ++ H->Size; H->Elements[i/2] > X; i = i/2) {
        H->Elements[i] = H->Elements[i/2] ; 
    }
    H->Elements[i] = X ;
}

(2)刪除最小元素(即根?結(jié)點)DeleteMin,下濾的方式。刪除最小元素操作的復雜度O(logN),因為刪除后涉及重建堆。

ElementType DeleteMin(PriorityQueue H) {
    int i, Child ;
    ElementType MinElement, LastElement;

    if( IsEmpty(H) ) {        //判斷堆是否為空
        Error( "Priority queue is empty" );
        return H->Elements[0];
    }
    MinElement = H->Elements[1];
    LastElement = H->Elements[H->Size--]; //在?根結(jié)點刪除了第一個堆結(jié)點,因此在根結(jié)點制造了一個“空穴”,
先保存最后一個堆結(jié)點,堆大小減1。下面的?代碼是把LastElement放到合適的地方。

    for( i =1; i*2 <= H->Size; i= Child) {  //下濾過程
        
        //找較小的一個兒子結(jié)點
        Child = i * 2;            //左兒子結(jié)點
        if(Child != H->Size && H->Elements[Child + 1] < H->Elements[Child] )
            Child++;               //如果左兒子不是新堆的最后一個結(jié)點,且左兒子大于右兒子,我們選擇右兒子
        
        //判斷較小的兒子結(jié)點是否小于LastElement。若是,把該較小的兒子結(jié)點上移到自己的父結(jié)點;
        //若不是,退出循環(huán)。(即LastElement比兒子結(jié)點都小,可以作為它們的父結(jié)點插入)
        if(LastElement > H->Elements[Child] ) 
            H->Elements[i] = H->Elements[Child];
        else
            break;
    }
    H->Elements[i] = LastElement ; 
    return MinElement;    //返回被刪除的根節(jié)點
}

(3)如果僅僅是?要獲得最小值,那么可以在常數(shù)時間完成O(1)。

</br>

最大堆:父結(jié)點的鍵值總是?大于或等于任何一個子節(jié)點的鍵值。

(1)插入操作Insert,采用制造“空穴”,上濾的方式——“空穴”在堆中不斷上移直到找出正確的位置。插入操作復雜度O(logN)。

void Insert(ElementType X, PriorityQueue H) {
    int i ;
    
    if( IsFull(H) ) {
        Error( "Priority queue is full" );
        return ;
    }

    for(i=++H->Size; H->Elements[i/2] < X; i = i/2) {
        H->Elements[i] = H->Elements[i/2];
    }
    H->Elements[i] = X;
}

對比最小堆的插入操作,可看到只是把for循環(huán)的條件">X"改為"<X"。意思是只要X比它的父結(jié)點大,就一直上濾。

(2)刪除最大元素(即根?結(jié)點)DeleteMax,下濾的方式。刪除最大元素操作的復雜度O(logN),因為刪除后涉及重建堆。

ElementType DeleteMax(PriorityQueue H) {
    int i, Child;
    ElementType MaxElement, LastElement;

    if( IsEmpty(H) ) {
        Error( "Priority queue is empty" );
        return H->Elements[0];
    }

    MaxElement = H->Elements[1];
    LastElement = H->Elements[H->Size--];

    for(i=1; 2*i <=H-Size; i = Child) {
        Child = i*2;
*        if(Child != H->Size && H->Elements[Child+1] > H->Elements[Child])
            Child++;

*        if(LastElement < H->Elements[Child] )
            H->Elements[i] = H->Elements[Child];
        else
            break;
    } 
    H->Elements[i] = LastElement;
    return MaxElement;
}

與最小堆的刪除最小元素操作相比,有2處條件判斷發(fā)生改變,已用*號標出,讀者可以自行體會。
(3)如果僅僅是要獲得最大值,那么可以在常數(shù)時間完成O(1)。

</br>

二叉堆的應用——堆排序:

用最大堆完成堆排序,每次DeleteMax花費時間O(logN),對N個元素進行排序就是O(NlogN)。
另外建立二叉堆花費的時間為O(N)。不解可參考->《為什么堆排序構(gòu)建堆的時間復雜度是N》
所以堆排序的時間復雜度為O(NlogN)+O(N) = O(NlogN),因為是就地排序,所以空間復雜度為O(1)。

void HeapAdjust(int *a, int i, int size) {   //調(diào)整堆
    int Child;
    int tmp;
    
    for(tmp = a[i]; 2*i+1 < size; i = Child) {
        Child = 2*i+1;   //左兒子
        if(Child != size-1 && a[Child+1] > a[Child])   //如果右兒子比左兒子大,取右兒子
            Child++;
    
        if(tmp < a[Child])
            a[i] = a[Child];
        else
          break;
    }
    a[i] = tmp;
}

void HeapSort(int *a, int size) {    //堆排序主例程
    int i;
    for(i = size/2; i>=0; i--) {    //構(gòu)建堆
        HeapAdjust(a, i, size);
    }

    for(i = size-1; i>0; i--) {
        int temp = a[i];   //交換堆頂和最后一個元素,即每次將剩余元素中的最大者放到最后面 
        a[i] = a[0];
        a[0] = temp;
        HeapAdjust(a, 0, i);  //重新調(diào)整堆頂節(jié)點成為大頂堆
    }
    
}

注:堆排序代碼的堆結(jié)點從數(shù)組索引0開始

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關(guān)閱讀更多精彩內(nèi)容

  • 我們有意調(diào)整了排序的順序,最后講這個堆排序。不是因為它很難,而是它涉及到了基本的數(shù)據(jù)結(jié)構(gòu)知識。 堆,又名“優(yōu)先隊列...
    吃個小燒餅閱讀 432評論 0 3
  • 二叉堆 堆有序定義:當一顆二叉樹的每個節(jié)點都大于等于它的兩個子節(jié)點時, 被稱為堆有序。二叉堆定義: 二叉堆是一組能...
    leoLy閱讀 2,109評論 0 2
  • 一、優(yōu)先隊列 1.簡單介紹 優(yōu)先隊列是一種抽象的數(shù)據(jù)結(jié)構(gòu),它與我們生活中的許多場景息息相關(guān)。比如我們的電腦或者手機...
    丶legend閱讀 994評論 0 0
  • 1.什么是二叉堆 二叉堆是一種特殊的堆,二叉堆是完全二元樹(二叉樹)或者是近似完全二元樹(二叉樹)。二叉堆有兩種:...
    RavenX閱讀 2,369評論 0 3
  • 什么是堆? 如圖所示:當父結(jié)點的值,都比其子結(jié)點的值小時,就是小根堆,一個小根堆中,最小的值肯定在堆頂。 同理,當...
    wayyyy閱讀 622評論 0 0

友情鏈接更多精彩內(nèi)容