堆排序代码c++
时间: 2023-06-21 08:20:16 浏览: 48
好的,以下是堆排序的C++代码:
```c++
#include <iostream>
using namespace std;
void heapify(int arr[], int n, int i) {
int largest = i;
int l = 2*i + 1;
int r = 2*i + 2;
if (l < n && arr[l] > arr[largest])
largest = l;
if (r < n && arr[r] > arr[largest])
largest = r;
if (largest != i) {
swap(arr[i], arr[largest]);
heapify(arr, n, largest);
}
}
void heapSort(int arr[], int n) {
for (int i = n / 2 - 1; i >= 0; i--)
heapify(arr, n, i);
for (int i=n-1; i>=0; i--) {
swap(arr[0], arr[i]);
heapify(arr, i, 0);
}
}
void printArray(int arr[], int n) {
for (int i = 0; i < n; ++i)
cout << arr[i] << " ";
cout << "\n";
}
int main() {
int arr[] = {12, 11, 13, 5, 6, 7};
int n = sizeof(arr)/sizeof(arr[0]);
cout << "Original array: ";
printArray(arr, n);
heapSort(arr, n);
cout << "Sorted array: ";
printArray(arr, n);
}
```
这段代码中包含了两个函数:`heapify` 和 `heapSort`。
`heapify` 函数将给定的数组中的元素进行堆化,即将元素按照一定规则构建成最大堆或最小堆。这里使用的是构建最大堆的方式。`heapify` 函数的时间复杂度为 O(log n)。
`heapSort` 函数则是堆排序的主要实现。首先,它将给定的数组进行堆化,然后将堆顶元素(即最大值或最小值)与堆底元素进行交换,并将堆大小减 1。接着,它再次对堆进行堆化,重复上述步骤直至堆为空。`heapSort` 函数的时间复杂度为 O(n log n)。
最后,`printArray` 函数用于打印排序后的数组。