体验百度文心一言AI模型生成堆排序C++版

以下是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集成开发环境

猜你喜欢

转载自blog.csdn.net/m0_38127487/article/details/132223125