堆排序

#include <stdio.h>

// 交换函数
void swap (int a[], int i, int j)
{
    int tmp = a[i];
    a[i] = a[j];
    a[j] = tmp;
}

// 打印数组
void printA (int *a, int len)
{
    int i;
    for (i = 0; i < len; i++)
    {
        printf ("%4d", a[i]);
    }
    
    printf ("\n");
}

// a 代表一个数组
// i 代表要调整的结点的下标
// len  数组的长度
void heapify(int *a, int i, int len)
{
    int left = 2 * i + 1;     // 左孩子结点下标
    int right = 2 * i + 2;    // 右孩子结点下标
    int max = i;     // 三个节点中最大元素的下标
    
    if (left < len && a[left] > a[max])
        max = left;
    if (right < len && a[right] > a[max])
        max = right;
    
    if (max != i)   // 当前父节点不是所有结点中最大的元素，需要做调整
    {
        swap (a, i, max);
        heapify (a, max, len);   // 调整被交换的结点
    }
}

void heapSort (int *a, int len)
{
    // 建堆
    int i;
    for (i = len/2 - 1; i >= 0; i--)
    {
        heapify (a, i, len);
    }
    
    // 排序
    for (i = len-1; i > 0; i--)
    {
        swap (a, 0, i);       // 拿堆顶元素与队尾元素进行交换
        len--;                // 找到一个最大元素以后堆大小减1
        heapify (a, 0, len);  // 调整堆顶元素
    }
}

int main()
{
    int a[10] = {9,6,8,0,3,5,2,4,7,1};
    int len = sizeof(a) / sizeof(a[0]);
    
    heapSort(a, len);
    
    printA (a, len);
    
    return 0;
}