#include <iostream> #include <vector> #include <cstdlib> using namespace std;
// 调整堆:将以i为根的子树调整为最大堆 // vec: 待排序数组, n: 堆的大小, i: 当前要调整的根节点索引 void heapify(vector<int>& vec, int n, int i) { int largest = i; // 初始化最大值为根节点 int left = 2 * i + 1; // 左子节点索引 (i的左孩子) int right = 2 * i + 2; // 右子节点索引 (i的右孩子)
// 如果左子节点大于根节点,更新最大值索引 if (left < n && vec[left] > vec[largest]) { largest = left; }
// 如果右子节点大于当前最大值,更新最大值索引 if (right < n && vec[right] > vec[largest]) { largest = right; }
// 如果最大值不是根节点,需要交换并递归调整受影响的子树 if (largest != i) { swap(vec[i], vec[largest]); // 递归调整交换后的子树,确保其仍为最大堆 heapify(vec, n, largest); } }
// 堆排序主函数 void heapSort(vector<int>& vec) { int n = vec.size();
// 1. 构建最大堆(从最后一个非叶子节点开始向前调整) // 最后一个非叶子节点的索引:n/2 - 1 for (int i = n / 2 - 1; i >= 0; i--) { heapify(vec, n, i); }
// 2. 逐个提取堆顶元素(最大值),放到数组末尾 for (int i = n - 1; i > 0; i--) { // 将当前堆顶(最大值)与堆的最后一个元素交换 swap(vec[0], vec[i]); // 对剩余的前i个元素重新调整为最大堆(堆大小变为i) heapify(vec, i, 0); } }
int main() { vector<int> vec = { 1,2,3,5,7,9,4,6,8 }; cout << "排序前的数组:"; for (int num : vec) { cout << num << " "; } cout << endl;
heapSort(vec);
cout << "排序后的数组:"; for (int num : vec) { cout << num << " "; } cout << endl;