一.堆的定义
堆就是以完全二叉树的顺序存储方式来存储元素,同时又要满足父亲结点存储数据都要大于儿子结点存储数据(也可以是父亲结点数据都要小于儿子结点数据)的一种数据结构。堆只有两种即大堆和小堆,大堆就是父亲结点数据大于儿子结点数据,小堆则反之。
二.堆的性质
- 1.双分支节点(有俩个儿子)比叶子节点(没有儿子)的数目少1
- 2.非叶子节点的下标位置为0到(n-2)/2
- 3.单分支节点(有一个儿子)没有或只有一个
- 4.左分支节点的位置=父节点的位置*2,右分支节点位置=父节点的位置*2+1,父节点位置=(任何一个孩子节点-1)/2。
其中3,4俩点都很好证明,我们来讨论下第1,2点。
在只有一个节点时,n2(代表双分支节点)=0,n1(单分支节点)=0,n0(叶子节点)=1。
当增加一个节点时,由于是完全二叉树只能是下面这样。此时n2=0,n1(单分支节点)=1,n0=1。
再增加一个节点时,n2=1,n1=0,n0=2
再继续往下推导,我们可以得出一个结论,当增加的节点让父节点成为单分支节点时,n0不变(父节点原先是n0,变成n1,但新增的子节点为n0), n2也不变,当增加的节点让父节点成为双分支节点时,n0加1,n2也加1,n0,n1同时增加,又因为n0初始为1,故n0=n2+1。我们还可以得出n1=1时节点的数目为偶数,n1=0时,节点的数目为奇数。
这里我们可以列出俩个方程
n2+n1+n0=n;
n0=n2+1;
能解出n0=(n+1-n1)/2;
故非叶子节点数组下标的末尾位置nx=n-n0-1=(n-3+n1)/2
当n为偶数n1=1,nx=(n-2)/2
当n为奇数时,n1=0,n-3是偶数,当我们对一个偶数+1后除2取整,和不加1除2取整值是一样大的,故nx=(n-3)/2=(n-2)/2,注意c语言整数除后要向下取整。
因此我们可以统一的认为非叶子节点的下标末尾位置为(n-2)/2.
三.代码实现
1.建堆:
在这里我们用向下建堆法,建一个大堆,向下建堆法就是确保每个非叶子节点都比子节点大(大堆是大,小堆是小),而非叶子节点的数组下标尾是(n-2)/2。
看代码:
void Heapsort(int* a, int n)
{
for (int i = (n - 2) / 2; i >= 0; i--)
{
AdjustDwonSort(a, n, i);
}
}
void AdjustDwonSort(int *a,int n,int parent)
{
int child = parent * 2 +1;
while (child < n)
{
if (child + 1 < n && a[child + 1] > a[child])
{
child++;
}
if (a[parent] < a[child])
{
Swap(a + parent, a + child);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
AdjustDwonSort函数作用是让比子节点小的父节点,与子节点中更大的子节点交换位置,这样我们通过确保每个非叶子节点都比子节点大从而成功建堆。
2.堆排:
思路:我们先建一个大堆,此时他的第一个元素是最大的,交换第一个元素和最后一个元素,这样最大的元素就在正确的位置,我们再用n-1个元素来调整重新建堆,再次和倒数第二个元素交换,这样循环往复,就能成功排序。
代码如下:
void AdjustDwonSort(int *a,int n,int parent)
{
int child = parent * 2 +1;
while (child < n)
{
if (child + 1 < n && a[child + 1] > a[child])
{
child++;
}
if (a[parent] < a[child])
{
Swap(a + parent, a + child);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
void Heapsort(int* a, int n)
{
for (int i = (n - 2) / 2; i >= 0; i--)
{
AdjustDwonSort(a, n, i);
}
for (int i = 0; i < n-1; i++)
{
Swap(a+0, a + n - i - 1);
AdjustDwonSort(a, n - 1 - i, 0);
}
}