堆排序

堆是一个完全二叉树的数组对象。树的每一层都是满的,最后一层可能除外(因为从一个节点的左子树开始填)。

例如

0-1-2-3-4(索引)
16-14-10-2-1(值)

节点位置计算

给定一个节点,可以很容易的计算父节点和子节点的位置。
parent(i)=floor((i-1)/2)
leftChild=2(i+1)-1
rightChild=2
(i+1)
注意:可以将乘2算法变成向左移位,除2算法变成向右移位。
即:
parent(i)=(i-1)>>1
leftChild=(i+1)<<1-1
rightChild=(i+1)<<1

堆排序

堆排序的时间复杂度为O(nlogn)是比较有效率的一种。其使用的是最大堆。最大堆的意思是父节点的值>=孩子节点的值。所以,堆排序每次循环把最大值移走,然后从剩下的节点重新建立最大堆。

第一步:

首先定义堆的结构体

typedef struct heap_t{
    int *arr;//point for an array to store heap value.
    int heapMaxIndex;//heap element max index number
    int arrLength;//array length of arr
} Heap;

其中arr指针指向的是存放堆数据的数组。
heapMaxIndex是堆的最大序号。比它小的都属于堆。
arrLength是数组最大的序号。如数组定义为a[10],那么arrLength的值应该为9。

第二步:

保持堆的性质
这一步是堆排序的基础。这里将功能写成一个函数名为void maxHeapify(Heap *hp, unsigned int nodei),这个函数用于让一个数组变成一个符合堆性质的数组。时间复杂度为O(h),h是堆所属二叉树的高度=logn(n是节点个数)。
思想:
从一个节点i,和它的孩子leftChild(i)和rightChild(i)中找到最大的,然后其索引存放在largest中。如果i的值是最大的。那么i为根的子树已经是最大堆,程序结束。
如果i的值不是最大的,那么将i的值和largest的值交换。下标为largest的节点在交换后作为父节点,那么它有可能违反堆的性质,因此递归调用该函数。

void maxHeapify(Heap *hp, unsigned int nodei) {
    unsigned int l = (nodei + 1) << 1 - 1;//left child
    unsigned int r = (nodei + 1) << 1;//right child
    unsigned int largest = 0;
    int heapMaxI = hp->heapMaxIndex;
    if(l <= heapMaxI && hp->arr[l] > hp->arr[nodei])
        largest = l;
    else
        largest = nodei;
    if(r <= heapMaxI && hp->arr[r] > hp->arr[largest])
        largest = r;

    if(largest!=nodei) {
        swap(&(hp->arr[largest]), &(hp->arr[nodei]));
        maxHeapify(hp, largest);
    } else
        return ;
}
第三步 利用maxHeapify函数创建堆

对于1个size为n的堆,我们可以分析到,n/2-1之前的都是父节点,其他的都是叶子节点,我们只需要对父节点进行maxHeapify就可以了。
n/2可以用右移运算n>>1。

Heap* createHeap(int* arrp,int  arrMaxIndex, int arrLength) {
    Heap* heap = malloc(sizeof(Heap));
    assert(heap!=NULL);
    heap->arr = arrp;
    heap->heapMaxIndex = arrMaxIndex;
    heap->arrLength = arrLength;
    int i;
    for(i = (arrMaxIndex + 1) >> 1 - 1; i >=0; i--)
        maxHeapify(heap, i);
    return heap;
}
第四步 堆排序

设堆的数组为A[0..n-1],调用maxHeapify函数就可以得到最大值,然后将最大值和n-1互换,把堆的大小heapMaxIndex减1,再次调用maxHeapify,又得到最大值,存放在A[0],再和A[n-2]互换,把堆的大小再减一,这样循环下去,直到堆的大小为0。那么我们就得到了从小到大的排好序的数组。

void heapSort(Heap* hp) {
    int last;
    while(hp->heapMaxIndex > 0) {
        last = hp->heapMaxIndex;
        swap(&(hp->arr[0]), &(hp->arr[last]));
        hp->heapMaxIndex -= 1;
        maxHeapify(hp, 0);
    }
}
第五步 测试堆排序
int main(void) {
    int array[] = {15,25,32,23,1,-4,35,2,-85,42,0,12,26,56,45,12,145,17,25,21};
    printArr(array, getSize(array));
    Heap* hp = createHeap(array, getSize(array)-1, getSize(array));
    heapSort(hp);
    printArr(array, getSize(array));
    freeHeap(hp);
    return 0;
}

完整的程序如下

#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
typedef struct heap_t{
    int *arr;//point for an array to store heap value.
    int heapMaxIndex;//heap element max index number
    int arrLength;//array length of arr
} Heap;
#define getSize(array) (sizeof array / sizeof *array)
Heap* createHeap(int* arrp,int  arrMaxIndex, int arrLength);
void swap(int* a, int* b);
void heapSort(Heap* hp);
void freeHeap(Heap* heap);
void maxHeapify(Heap *hp, unsigned int nodei);
void printArr(int *p, int size);
int main(void) {
    int array[] = {15,25,32,23,1,-4,35,2,-85,42,0,12,26,56,45,12,145,17,25,21};
    printArr(array, getSize(array));
    Heap* hp = createHeap(array, getSize(array)-1, getSize(array));
    heapSort(hp);
    printArr(array, getSize(array));
    freeHeap(hp);
    return 0;
}

void swap(int* a, int* b) {
    int tmp;
    tmp = *a;
    *a = *b;
    *b = tmp;
}

Heap* createHeap(int* arrp,int  arrMaxIndex, int arrLength) {
    Heap* heap = malloc(sizeof(Heap));
    assert(heap!=NULL);
    heap->arr = arrp;
    heap->heapMaxIndex = arrMaxIndex;
    heap->arrLength = arrLength;
    int i;
    for(i = (arrMaxIndex + 1) >> 1 - 1; i >=0; i--)
        maxHeapify(heap, i);
    return heap;
}

void heapSort(Heap* hp) {
    int last;
    while(hp->heapMaxIndex > 0) {
        last = hp->heapMaxIndex;
        swap(&(hp->arr[0]), &(hp->arr[last]));
        hp->heapMaxIndex -= 1;
        maxHeapify(hp, 0);
    }
}

void freeHeap(Heap* heap) {
    free(heap);
}

void maxHeapify(Heap *hp, unsigned int nodei) {
    unsigned int l = (nodei + 1) << 1 - 1;//left child
    unsigned int r = (nodei + 1) << 1;//right child
    unsigned int largest = 0;
    int heapMaxI = hp->heapMaxIndex;
    if(l <= heapMaxI && hp->arr[l] > hp->arr[nodei])
        largest = l;
    else
        largest = nodei;
    if(r <= heapMaxI && hp->arr[r] > hp->arr[largest])
        largest = r;

    if(largest!=nodei) {
        swap(&(hp->arr[largest]), &(hp->arr[nodei]));
        maxHeapify(hp, largest);
    } else
        return ;
}
void printArr(int *p, int size) {
    int i;
    for(i = 0; i < size; i++) {
        printf("%d ", p[i]);
    }
    printf("\n");
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,937评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,503评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,712评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,668评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,677评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,601评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,975评论 3 396
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,637评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,881评论 1 298
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,621评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,710评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,387评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,971评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,947评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,189评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,805评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,449评论 2 342

推荐阅读更多精彩内容