堆排序

堆排序的思想(以最小堆为例):

  • 给的无序数列,首先要将数组改造成符合最小堆
    • 关于最小堆的建造,从最后一个非叶结点开始,每个结点都向下调整
  • 若要增加记录,应该加在数组的最后,然后对这个数向上调整,知直到找到合适的位置
  • 若要删除最小值,则只要将第一个与最后一个互换,数组长度减一,从堆顶向下调整
  • 若要排序,则只要从最后一个数开始向前遍历,遍历到的每个数与堆顶互换后从堆顶向下调整,调整时数组末尾已排序好的数字不计入操作
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

void heapDown(int nums[], int i, int n) {
    
    int temp = nums[i];
    int j = 2 * i + 1;
    while (j < n) {
        if (j + 1 < n && nums[j + 1] < nums[j]) {
            j = j + 1;
        }
        if (temp < nums[j]) break;
        nums[i] = nums[j];
        i = j;
        j = 2 * i + 1;
    }
    nums[i] = temp;
}

void heapUp(int nums[], int i) { //向上调整只有在增加数的时候用到
    int temp = nums[i];
    int j = (i - 1) / 2;
    while (j >= 0 && i != 0) {
        if (nums[j] <= temp) { //只需比较新加入的数与父结点
            break;
        }
        nums[i] = nums[j];
        i = j;
        j = (i - 1) / 2;
    }
    nums[i] = temp;
}

void createHeap(int nums[], int n) {
    int i;
    for (i = (n - 1) / 2; i >= 0; i--) { //从最后一个非叶子结点开始向下调整
        heapDown(nums, i, n);
    }
}

void deleteRootNum(int nums[], int n) {
    swap(nums[0], nums[n - 1]);
    heapDown(nums, 0, n - 1);
}

void addNum(int nums[], int n, int num) {
    nums[n-1] = num;
    heapUp(nums, n-1);
}

void heapSorted(int nums[], int n) {
    for (int i = n - 1; i >= 0; i--) {
        swap(nums[i], nums[0]);
        heapDown(nums, 0, i);
    }
}

void print(string str, int nums[], int n) {
    cout<

你可能感兴趣的:(堆排序)