1.堆的定义
该数组从逻辑上讲就是一个堆结构,我们用简单的公式来描述一下堆的定义就是:
大顶堆:arr[i] >= arr[2i+1] && arr[i] >= arr[2i+2]
小顶堆:arr[i] <= arr[2i+1] && arr[i] <= arr[2i+2]
2.形象表示:
3.建立一个堆
步骤一 构造初始堆。将给定无序序列构造成一个大顶堆(一般升序采用大顶堆,降序采用小顶堆)。
a.假设给定无序序列结构如下
此时我们从最后一个非叶子结点开始(叶结点自然不用调整,第一个非叶子结点 i=arr.length/2-1=4/2-1=1,也就是下面的8结点),从左至右,从下至上进行调整。这个过程调用函数maxHeap(array,i)即可
然后用for循环使i--,不断循环调用即可,所以这一部分的函数为:
void buildMaxHeap(int array[],int length)
{
int i;
for(i=(length)/2-1;i>=0;i--)
maxHeapify(array,i);
}
4.找到第二个非叶节点4,由于[4,9,8]中9元素最大,4和9交换。
这时,交换导致了子根[4,5,6]结构混乱,继续调整,[4,5,6]中6最大,交换4和6。
此时,我们就将一个无需序列构造成了一个大顶堆。
4.实现 maxHeapify(array,i)函数
int temp = arr[i];//先取出当前元素i
for(int k=i*2+1;k<length;k=k*2+1){//从i结点的左子结点开始,也就是2i+1处开始
if(k+1<length && arr[k]<arr[k+1]){//如果左子结点小于右子结点,k指向右子结点
k++;
}
if(arr[k] >temp){//如果子节点大于父节点,将子节点值赋给父节点(不用进行交换)
arr[i] = arr[k];
i = k;
}else{
break;
}
}
arr[i] = temp;//将temp值放到最终的位置
5.堆顶元素分别置换到数组的尾部,并重新整堆
for(i=n-1;i>=0;i--)
{
swap(&arr[0],&arr[i]);
maxHeapify(arr,0,i);
}
6 程序:
#include<iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
void print(int a[],int n)
{
char i=0;
for(i=0;i<n;i++)
printf("%d\t",a[i]);
printf("\n");
}
void swap(int *a,int *b)
{
int temp=*a;
*a=*b;
*b=temp;
}
void maxHeapify(int arr[],int i,int length)
{
int temp = arr[i];//先取出当前元素ifor(int k=i*2+1;k<length;k=k*2+1){//从i结点的左子结点开始,也就是2i+1处开始
if(k+1<length && arr[k]<arr[k+1]){//如果左子结点小于右子结点,k指向右子结点
k++;
}
if(arr[k] >temp){//如果子节点大于父节点,将子节点值赋给父节点(不用进行交换)
arr[i] = arr[k];
i = k;
}else{
break;
}
}
arr[i] = temp;//将temp值放到最终的位置
}
void buildMaxHeap(int array[],int length)
{
int i;
for(i=((length)/2-1);i>=0;i--)
maxHeapify(array,i,length);//整堆
}
void heap_sort(int arr[],int n)
{
int i;
buildMaxHeap(arr,n);
for(i=n-1;i>=0;i--)
{
swap(&arr[0],&arr[i]);
maxHeapify(arr,0,i);
}
}
int main()
{
int a[]={4,6,8,5,9};
printf("a:\n");
print(a,5);
printf("after sort:\n");
heap_sort(a,5);
print(a,5);
return 0;
}