以下是C++实现的堆排序算法:
#include <iostream>
#include <vector>
using namespace std;
void heapify(vector<int>& nums, int n, int i) {
int largest = i; // 初始化最大值为根节点
int l = 2 * i + 1; // 左子节点的下标
int r = 2 * i + 2; // 右子节点的下标
// 如果左子节点比根节点大,更新最大值为左子节点
if (l < n && nums[l] > nums[largest]) {
largest = l;
}
// 如果右子节点比最大值大,更新最大值为右子节点
if (r < n && nums[r] > nums[largest]) {
largest = r;
}
// 如果最大值不是根节点,进行交换并递归调整堆
if (largest != i) {
swap(nums[i], nums[largest]);
heapify(nums, n, largest);
}
}
void heapSort(vector<int>& nums) {
int n = nums.size();
// 构建最大堆(从最后一个非叶子节点开始调整)
for (int i = n / 2 - 1; i >= 0; i--) {
heapify(nums, n, i);
}
// 依次将最大值(根节点)交换到末尾,并重新调整堆
for (int i = n - 1; i >= 0; i--) {
swap(nums[0], nums[i]);
heapify(nums, i, 0);
}
}
int main() {
vector<int> nums = {5, 2, 8, 3, 1, 6};
heapSort(nums);
for (int num : nums) {
cout << num << " ";
}
cout << endl;
return 0;
}
堆排序算法的核心是构建最大堆和调整堆。首先从最后一个非叶子节点开始,
对每个子树进行堆化操作,构建最大堆。然后依次将最大值(根节点)交换到末尾,
并重新调整堆,直到排序完成。
C:\Users\Administrator\CLionProjects\untitled1\cmake-build-debug\untitled1.exe
1 2 3 5 6 8
Process finished with exit code 0
开发工具:百度文心一言
CLion集成开发环境