堆排序

来源:互联网 发布:mysql 监控 编辑:程序博客网 时间:2024/05/29 10:24

     一 初识堆

堆 数据结构是一种数组,它可以视为一颗完全二叉树。如下图:

        

图中的树是数组,A={16, 14, 10, 8, 7, 9, 3, 7},圈内表示数值,圈外红色的数字表示数组的下标。

array_size是数组的大小(此时是8),heap_size是构建堆的元素的多少。满足heap_size<= array_size

给定某结点的下表i,其父结点下标为PARENT(i), 左儿子下标为LEFT(i), 右儿子下标为RIGHT(i)。满足

         PARENT(i) = int((i-1)/2) ;   LEFT(i) = 2*i+1;   RIGHT(i)=2*i+2

最大堆满足:对于任何结点(除根节点)PARENT(i) > i

最小堆满足:对于任何结点(除根节点)PARENT(i) <i

叶子节点(下标):int(i/2), int(i/2)+1......array_size

下面的介绍以最大堆为例

                                              二 保持堆的性质

       最大堆的性质为 PARENT(i) > i,因此对于特定的结点,应满足比左右儿子都大

    在上图中下标为1的结点值为2,左孩子为4,右孩子为1,1结点比左孩子小,就让1结点和3结点数值换过来。若此时4结点大于4,就把1结点和4结点数值换过来。

参考程序:

复制代码
void MAX_HEAPIFY(int *A, int heap_size, int i)   //i 为待处理保持性质的结点下标{        int l = 2 * i + 1;                       //左孩子        int r = 2 * i + 2;                       //右孩子        int largest = i;        if(l < heap_size && A[largest] < A[l])   // 左孩子数值大        {            largest = l;        }     if(r < heap_size && A[largest] < A[r])   //右孩子数值大
{ largest
= r; } if(largest != i) { int tmp = A[i]; A[i] = A[largest]; A[largest] = tmp; MAX_HEAPIFY(A, heap_size, largest); //再从数值大的结点继续往下递归处理 }}
复制代码

                                              三 构建堆

      从底向上使得每个结点保持堆的性质就可以构建堆,因为叶子节点就自己,无儿女,因此从倒数第一个非叶子节点开始依次往上构建,直到树根位置,而叶子节点开始的位置是int(i/2),因此第一个令其保持性质的结点下标为int((i-1)/2)。接着是int((i-1)/2-1)一直到0为止。下图表示了构建堆的详细过程:

图中有9个结点,int(9/2)= 4, 从4-1=3开始(依次是4 3 2 1 0)。注意是从后往前,为何不是从前往后,显然如果是上图中,从0开始,是6和3交换,我们知道在最大堆中根是最大的,可是从前往后的话,最大值7在无出头之日。

参考程序:

复制代码
void BUILD_MAX_HEAP(int *A, int array_size, int heap_size){        int i;        for (i=array_size/2-1; i>=0; i--)        {            MAX_HEAPIFY(A, heap_size, i);        }}
复制代码

                                              四 堆排序

      最大堆把最大值排到了首个位置A[0],这是如果把最大值和最后一个值A[heap_size-1]换过来,再使A[0]保持堆的性质,再使heap_size自建。重复以上过程,就是个非递减排序。下图是一个事例过程:

参考程序:

复制代码
void HEAP_SORT(int *A, int array_size, int heap_size){        int tmp;        BUILD_MAX_HEAP(A, array_size, heap_size);        while(heap_size > 1)        {            tmp = A[0];            A[0] = A[heap_size-1];            A[heap_size] = tmp;            heap_size--;            MAX_HEAPIFY(A, heap_size, 0);        }}
复制代码

测试

复制代码
#include <iostream>#include <vector>using namespace std;void maxHeapify(int A[], int lens, int i){    if (A == NULL || i >= lens)        return;    int l = 2 * i + 1;    int r = 2 * i + 2;    int largest = i;    if (l < lens || A[i] > A[largest])        largest = l;    if (r < lens && A[r] > A[largest])        largest = r;    if (largest != i)    {        int tmp = A[largest];        A[largest] = A[i];        A[i] = tmp;        maxHeapify(A, lens, largest);    }}void buildHeap(int A[], int lens){    if (A == NULL || lens <= 0)        return;    int mid = (lens-2) / 2;    for (int i = mid; i >= 0; --i)        maxHeapify(A, lens, i);}void heapSort(int A[], int lens){    buildHeap(A, lens);    while (lens > 1)    {        int tmp = A[lens-1];        A[lens-1] = A[0];        A[0] = tmp;        --lens;        maxHeapify(A, lens, 0);    }}void tranverse(int A[], int lens){    if (A == NULL || lens <= 0)        return;    for (int i = 0; i < lens; ++i)        cout << A[i] << "  ";    cout << endl;}    int main(){    int A[] = {3, 2, 8, 7, 0, 11};    int lens = sizeof(A) / sizeof(*A);    tranverse(A, lens);//    buildHeap(A, lens);//    tranverse(A, lens);    heapSort(A, lens);    tranverse(A, lens);}
复制代码

                                            

五 堆操作

1.HEAP_MAXIMUM(A) 找出最大的元素

int  HEAP_MAXIMUM(int *A){        return A[0];}

2.int HEAP_EXTRACT_MAX(int *A, int heap_size) 找出堆中最大元素,并删除,保持堆

复制代码
int HEAP_EXTRACT_MAX(int *A, int heap_size){        int max;        max = A[0];        A[0] = A[heap_size-1];        heap_size--;        MAX_HEAPIFY(A, heap_size, 0);                return max;}
复制代码

3. void HEAP_INCREASE_KEY(int *A, int heap_size, int i, int key) 第i个位置出若key比原来大,就改成key,保持堆

复制代码
void HEAP_INCREASE_KEY(int *A, int heap_size, int i, int key){        int tmp;        if(A[i] > key)        {            printf("The key is smaller than A[i]");        }        else        {            A[i] = key;            while(i>=1 && A[(i-1)/2] < A[i])            {                tmp = A[i];                A[i] = A[(i-1)/2];                A[(i-1)/2] = tmp;                i = (i-1)/2;                MAX_HEAPIFY(A, heap_size, i);            }        }}
复制代码

4.void MAX_HEAP_INSERT(int *A, int array_size, int heap_size, int key) 堆中插入元素

复制代码
void MAX_HEAP_INSERT(int *A, int array_size, int heap_size, int key){        array_size++;        heap_size++;        A[heap_size] = -32768;        HEAP_INCREASE_KEY(A, heap_size, heap_size-1, key);}
0 0