1.思想:
1.1 首先,大根堆的性质决定,大根堆的第一个元素是最大的。
1.2 然后,将大根堆第一个位置的元素取出,将最后一个元素赋给第一个,对前(n-1)个元素进行大根堆调整,将得到一个新的大根堆。
1.3 再取出第一个元素,将大根堆中的最后一个元素即数组中的第(n-1)个元素赋给第一个位置。对前(n-2)个元素进行大根堆调整。
1.4 ······
2.实现:
提供了两种进行大根堆调整的方法,我更喜欢递归的方法。
// 堆排序
/*
大根堆:根大于 左右孩子; 小根堆:跟小于左右孩子
此部分代码目的:堆排序
*/
#include<iostream>
using namespace std;
void swap(int& temp1, int& temp2)
{
int temp = temp1;
temp1 = temp2;
temp2 = temp;
}
// 创建大根堆 的方法1
void headAdjust1(int* a,int k,int len)
{
a[0] = a[k];
for (int i = 2 * k; i <= len; i *= 2)
{
if (i < len && a[i] < a[i + 1])
{
i++;
}
if (a[0] >= a[i])
break;
else
{
a[k] = a[i];
k = i;
}
}
a[k] = a[0];
}
// 创建大根堆的 方法2 // 使用递归
void headAdjust(int* a, int k, int len)
{
int i = 2 * k;
if (i > len)
{
return;
}
else if (i == len)
{
if (a[i] > a[k])
{
swap(a[i], a[k]);
}
}
else if (i < len)
{
if (a[i + 1] > a[i])
{
i++;
}
if (a[i] > a[k])
{
swap(a[i], a[k]);
}
headAdjust(a, i, len);
}
}
// 创建大根堆
void buildmaxHeap(int* a, int len)
{
for (int i = len / 2; i > 0; i--)
{
headAdjust(a, i, len);
}
}
//
void heapSort(int* a, int len)
{
for (int i = len; i > 1; i--)
{
swap(a[i], a[1]);
headAdjust(a, 1, i-1);
}
}
int main()
{
int a[] = { 0,31,23,97,64,108,14,88,2,34,6 };
int length = sizeof(a) / sizeof(a[0]) - 1;
cout << "生成大根堆" << endl;
buildmaxHeap(a, length);
for (int i = 1; i <= length; i++)
{
cout << a[i] << " ";
}
cout << endl;
cout << "堆排序" << endl;
heapSort(a, length);
for (int i = 1; i <= length; i++)
{
cout << a[i] << " ";
}
cout << endl;
system("pause");
return 0;
}
3.输出: