堆排序

#include <cstdio>

int main() {
    void HeapSort(int *a, int len);
    const int MAXLEN = 10;
    int a[MAXLEN] = {3, 8, 9, 1, 0, 2, 5, 7, 4, 6};

    printf("befor sort:");
    for(int i=0; i<MAXLEN; ++i) {
        printf("%d ", a[i]);
    }
    putchar('\n');

    HeapSort(a, MAXLEN);

    printf("after sort:");
    for(int i=0; i<MAXLEN; ++i) {
        printf("%d ", a[i]);
    }
    putchar('\n');

    return 0;
}

void SwapInt(int *e1, int *e2) {
    int tmp;
    tmp = *e1;
    *e1 = *e2;
    *e2 = tmp;
}

#define PARENT(i)  ((i+1)>>1) - 1 
#define LEFT(i)    ((i+1)<<1) - 1
#define RIGHT(i)   (i+1)<<1

void Heapify(int *a, int index, int len) {
    int left = LEFT(index);
    int right = RIGHT(index);

    int max = index;
    if (left<len && a[left]>a[max]) max = left;
    if (right<len && a[right]>a[max]) max = right;

    if (max != index) {
        SwapInt(&a[max], &a[index]);
        Heapify(a, max, len);
    }
}

void BuildHeap(int *a, int len) {
    for (int i=(len-1)>>1; i>=0; --i) {
        Heapify(a, i, len);
    }
}

void HeapSort(int *a, int len) {
    BuildHeap(a, len);

    for (int i=len-1; i>0; --i) {
        SwapInt(&a[0], &a[i]);
        Heapify(a, 0, i);
    }
}

要点:
1、使用连续的空间来实现“堆”(堆是完全二叉树),这样的计算 PARENT LEFT RIGHT 会方便
2、 理解Heapify 函数, 三数比较至少需要两次,还有递归的使用
3、理解BuilHeap 使用 (len-1)>>1, 因为可以证明区间( (len-1)>>1, len )所有的元素都是叶子
4、理解什么大顶堆、小顶堆的定义

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值