#include <iostream>
#include <vector>
using namespace std;
// 调整堆,使其满足堆性质,即父节点的值大于等于子节点的值
void heapify(vector<int>& array, int n, int i) {
int largest = i; // 初始化最大值为父节点
int left = 2 * i + 1; // 左子节点
int right = 2 * i + 2; // 右子节点
if (left < n && array[left] > array[largest]) {
largest = left;
}
if (right < n && array[right] > array[largest]) {
largest = right;
}
if (largest != i) { // 如果最大值不等于父节点,则交换位置,并递归调整子树
swap(array[i], array[largest]);
heapify(array, n, largest);
}
}
void heapSort(vector<int>& array) {
int n = array.size();
// 建立初始堆,从第一个非叶子节点开始调整堆
for (int i = n / 2 - 1; i >= 0; i--) {
heapify(array, n, i);
}
// 逐个从堆顶取出元素,放到数组末尾,并重新构建堆
for (int i = n - 1; i >= 0; i--) {
swap(array[0], array[i]);
heapify(array, i, 0);
}
}
int main() {
vector<int> array = {9, 5, 2, 7, 1, 6, 3};
heapSort(array);
for (int i = 0; i < array.size(); i++) {
cout << array[i] << " ";
}
return 0;
}
这是一个使用C++编写的高效的堆排序算法。堆排序利用了二叉堆的性质,通过构建最大堆(或最小堆),将最大(最小)元素置于数组的末尾,然后缩小堆的范围,并重复此过程,直到整个数组有序。
在heapify
函数中,通过比较父节点与其左右子节点的值,找出其中最大的一个,并进行交换,然后递归调整子树。在heapSort
函数中,首先建立初始堆,然后逐个取出堆顶元素,并重新构建堆,直到所有元素有序。
在主函数中,我们示例了一个包含一些无序元素的数组,并对其进行堆排序,并输出排序后的结果。
该算法的时间复杂度为O(nlogn),空间复杂度为O(1),其中n为输入数组的大小。