#include <iostream>
// 堆调整(非递归版)
void max_heapify_no_recur(int a[], int len, int i)
{
int l = 0;
int r = 0;
int m = 0;
while (true) {
l = i * 2 + 1; //左节点
r = i * 2 + 2; //右节点
m = i;
// 找出最大值节点
if (l < len && a[l] > a[m]) {
m = l;
}
if (r < len && a[r] > a[m]) {
m = r;
}
if (m != i) {
// 把当前节点设为最大,继续调整子节点
std::swap(a[i], a[m]);
i = m;
}
else {
break;
}
}
}
// 堆调整(递归版)
void max_heapify(int a[], int len, int i)
{
int l = i * 2 + 1; //左节点
int r = i * 2 + 2; //右节点
int m = i;
// 找出最大值节点
if (l < len && a[l] > a[m]) {
m = l;
}
if (r < len && a[r] > a[m]) {
m = r;
}
if (m != i) {
// 把当前节点设为最大,递归调整子节点
std::swap(a[i], a[m]);
max_heapify(a, len, m);
}
}
// 建堆
void build_max_heap(int a[], int len)
{
for (int i = len/2 - 1; i >= 0; i--) {
// max_heapify(a, len, i);
max_heapify_no_recur(a, len, i);
}
}
// 堆排序
void heap_sort(int a[], int len)
{
build_max_heap(a, len);
for (int i = len - 1; i > 0; i--) {
std::swap(a[i], a[0]);
// max_heapify(a, i, 0);
max_heapify_no_recur(a, i, 0);
}
}
int main(int argc, const char * argv[])
{
int a[] = {1, 1, 6, 6, 8, 3, 5, 100, 300, 200, 99, 99};
heap_sort(a, sizeof(a)/sizeof(a[0]));
for (int i : a) {
std::cout << i << ' ';
}
std::cout << std::endl;
return 0;
}
堆排序-C++代码
原文作者:CharlesZhangCh
原文地址: https://www.jianshu.com/p/0737072eca57
本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
原文地址: https://www.jianshu.com/p/0737072eca57
本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。