//代码如下
#include<iostream>
using namespace std;
//调整大顶堆(仅是调整过程,建立在大顶堆已构建的基础上)
void adjustHeap(int *arr, int i, int len)
{
int temp = arr[i]; //先取出当前元素i
for (int k = i * 2 + 1; k<len; k = k * 2 + 1) { //从i结点的左子结点开始,也就是2i+1处开始
if (k + 1<len && arr[k]<arr[k + 1]) { //如果左子结点小于右子结点,k指向右子结点
k++;
}
if (arr[k] >temp) { //如果子节点大于父节点,将子节点值赋给父节点(不用进行交换)
arr[i] = arr[k];
i = k;
}
else {
break;
}
}
arr[i] = temp;//将temp值放到最终的位置
}
void swap(int &a, int &b)
{
int temp = a;
a = b;
b = temp;
}
void heapSort(int *arr, int len)
{
//1、构建大顶堆
for (int i = len / 2 - 1; i >= 0; i--) {
//从第一个非叶子结点从下至上,从右至左调整结构
adjustHeap(arr, i, len);
}
//调整堆结构+交换堆顶元素与末尾元素
for (int j = len - 1; j > 0; j--) {
swap(arr[0], arr[j]); //将堆顶元素与末尾元素进行交换
adjustHeap(arr, 0, j);//重新对堆进行调整
}
}
int main()
{
int arr[]= { 9,8,7,6,10,5,4,3,2,1};
int len = sizeof(arr) / sizeof(int);
heapSort(arr, len);
for (int i = 0; i < len; i++) {
cout << arr[i] << " ";
}
cout << endl;
return 0;
}
图解堆排序
最新推荐文章于 2022-12-13 18:23:28 发布