如何在堆排序中使用用户输入数组

Question

我是一个初学者，如果这是一个愚蠢的问题，请原谅。我正在尝试编写一个堆排序，它以数组的形式接受用户的输入，并对其进行堆排序，但我无法在我的代码中找到实现用户输入数组的方法。最后，在main函数中，我必须使用数组arr[]={}，但是如果我将数组定义为arr10={}，则必须定义元素或元素的数量，以便创建零。下面是代码：

// Heap Sort in C++

#include <iostream>
using namespace std;

void heapify(int arr[], int n, int i) {
// Find largest among root, left child and right child
int largest = i;
int left = 2 * i + 1;
int right = 2 * i + 2;

if (left < n && arr[left] > arr[largest])
    largest = left;

if (right < n && arr[right] > arr[largest])
    largest = right;

// Swap and continue heapifying if root is not largest
if (largest != i) {
    swap(arr[i], arr[largest]);
    heapify(arr, n, largest);
}
}

// main function to do heap sort
void heapSort(int arr[], int n) {
// Build max heap
for (int i = n / 2 - 1; i >= 0; i--)
    heapify(arr, n, i);

// Heap sort
for (int i = n - 1; i >= 0; i--) {
    swap(arr[0], arr[i]);

    // Heapify root element to get highest element at root again
    heapify(arr, i, 0);
}
}

// Print an array
void printArray(int arr[], int n) {
for (int i = 0; i < n; ++i)
    cout << arr[i] << " ";
cout << "\n";
}

// Driver code
int main() {

int arr[10] = { };
cout << "Enter the numbers : " << endl;
for (int x = 1; x <= 5; x++)
{
    cin >> arr[x];
}
int n = sizeof(arr) / sizeof(arr[0]);
heapSort(arr, n);

cout << "Sorted array is \n";
printArray(arr, n);
}

下面是输出：code output image

Answer 1

您可以考虑std::vector并通过引用将其传递给heapify(...)以节省内存。