【數(shù)據(jù)結(jié)構(gòu)】堆排序(C++實現(xiàn))

術(shù)語解釋:

  • 大根堆:樹中所有非終端節(jié)點的值,不小于左右孩子節(jié)點的完全二叉樹
  • 建初堆:將r[1..n]調(diào)整為大根堆
  • 調(diào)整堆:去掉堆頂元素后,將剩余元素調(diào)整成一個新的大根堆

代碼實現(xiàn):

include <iostream>

using namespace std;

void swap(int &a, int &b);

void swap(int &a, int &b)
{
int t = a;
a = b;
b = t;
}

void HeapSort(int *a, int n);
void CreateHeap(int *a, int n);
void AdjustHeap(int *a, int s, int m);

void AdjustHeap(int *a, int s, int m)
{
int temp = a[s]; //取出堆頂元素
for (int i = 2 * s; i <= m; i *= 2) //i從s的左孩子開始
{
if (i<m && a[i] < a[i + 1]) //i為大孩子的位置,i<m表示有有孩子
{
i++;
}
if (temp >= a[i]) //堆頂元素大于倆孩子,說明已經(jīng)是堆了,結(jié)束
{
break;
}
a[s] = a[i]; //大孩子上去
s = i; //s為大孩子上去之前的位置
}

//現(xiàn)在已經(jīng)得到了s,也就是堆頂元素的正確位置
a[s] = temp; //將堆頂元素丟進(jìn)去

}

void CreateHeap(int *a, int n)
{
for (int i = n / 2; i > 0; i--)
{
AdjustHeap(a, i, n);
}
}

void HeapSort(int *a, int n)
{
CreateHeap(a, n); //建初堆
for (int i = n; i > 1; i--) //i一開始為最后一個元素的位置
{
swap(a[1], a[i]); //堆頂和最后一個元素交換
AdjustHeap(a, 1, i - 1); //將a[1..i-1]調(diào)整為大根堆
}
}

int main()
{
//下標(biāo)從1開始
int a[9] = {-1, 45, 34, 78, 12, 34, 32, 29, 64};
int b[9] = {-1, 12, 29, 32, 34, 34, 45, 64, 78};

int n = 8;
HeapSort(a, n);

bool sorted = true;
for (int i = 1; i <= n; i++)
{
    if (a[i] != b[i])
    {
        sorted = false;
        break;
    }
}

if (sorted)
{
    cout << "true" << endl;
}

//打印排序后的數(shù)組
for (int i = 1; i <= n; i++)
{
    cout << a[i] << " ";
}

cout << system("pause");
return 0;

}


參考資料:《數(shù)據(jù)結(jié)構(gòu)(C語言版)》p221-p225

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

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

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