时间复杂度:O(nlogn)
空间复杂度:O(1)
#include <iostream>
#include <stack>
#include <queue>
#include <vector>
#include <string>
using namespace std;
//堆的存储一般用数组来实现。
//假如父节点的数组下标为i的话,那么其左右节点的下标分别为:(2*i + 1)和(2*i + 2)。
//完全二叉树中,假如有n个元素,那么在堆中最后一个非叶子节点位置为(n / 2 - 1)。
//调整堆
void adjust(vector<int>& arr, int len, int root) {
int Lchild = 2 * root + 1;
int Rchild = 2 * root + 2;
int MaxIndex = root;
if (Lchild < len && arr[Lchild] > arr[MaxIndex])
MaxIndex = Lchild;
if (Rchild < len && arr[Rchild] > arr[MaxIndex])
MaxIndex = Rchild;
if (MaxIndex != root) {
swap(arr[MaxIndex], arr[root]);
adjust(arr, len, MaxIndex);
}
}
//堆排序
void HeapSort(vector<int>& arr, int size) {
//从最后一个非叶子节点向上构建大根堆
for (int i = size / 2 - 1; i >= 0; i--) {
adjust(arr, size, i);
}
//调整大根堆 j代表数组的长度,上面长度为size的大根堆已经构建好了
for (int j = size - 1; j >= 1; j--) { //现在开始构建长度为size-1的大根堆
swap(arr[0], arr[j]); //首先把堆顶元素(最大的元素)换到堆底
adjust(arr, j, 0); //重新调整堆的结构
}
}
int main() {
vector<int> arr = { 3, 1, 18, 4, 53, 24, 8, 10 };
HeapSort(arr, arr.size());
for (int i = 0; i < arr.size(); i++) {
cout << arr[i]<<" ";
}
return 0;
}