堆排序的实现

/*****************************************************头文件**************************************/

#pragma once

//调整堆、从第i个元素开始调整，整个堆的长度为iLength。
void HeapModify(int A[], int i, int iLength);

//建堆，从数组中间，往前，全部调整，则建立了堆。
void BuildHeap(int A[], int iLength);

//堆排序，先建堆，再将堆中最大元素、与最后一个元素互换、
//故最后一个元素最大，再重新调整堆，再互换。
void HeapSort(int A[], int iLength);

/*****************************************************头文件**************************************/

/*****************************************************实现文件**************************************/
#include "HeapSort.h"


//调整堆、从第i个元素开始调整，整个堆的长度为iLength。
void HeapModify(int A[], int i, int iLength)
{    
    //本身、左孩子，有孩子，看谁最大。
    int iMaxIndex = i;

    //A[i]的左孩子
    int iLeftChild = 2 * i + 1;
    if (iLeftChild < iLength && A[iLeftChild] > A[i])
    {
        iMaxIndex = iLeftChild;
    }

    int iRightChild = 2 * i + 2;
    if ( iRightChild < iLength && A[iRightChild] > A[iMaxIndex])
    {
        iMaxIndex = iRightChild;
    }

    //看是不是要交换元素、交换元素后，调整子树。
    if ( iMaxIndex != i)
    {
        int iValue = A[i];
        A[i] = A[iMaxIndex];
        A[iMaxIndex] = iValue;

        HeapModify(A,iMaxIndex, iLength);
    }

}

//建堆，从数组中间，往前，全部调整，则建立了堆。
void BuildHeap(int A[], int iLength)
{
    for ( int i = iLength/2 - 1; i >= 0; i --)
    {
        HeapModify(A, i, iLength);
    }
}


//堆排序，先建堆，再将堆中最大元素、与最后一个元素互换、
//故最后一个元素最大，再重新调整堆，再互换。
void HeapSort(int A[], int iLength)
{
    BuildHeap(A,iLength);
    for (int i = iLength - 1; i >= 0; i--)
    {
        int iEnd = A[i];
        A[i] = A[0];
        A[0] = iEnd;

        HeapModify(A,0, i);
    }
}
/*****************************************************实现文件**************************************/

/*****************************************************测试主函数文件**************************************/
#include "stdafx.h"
#include <iostream>
#include <cstdlib>
#include <ctime>
#include "HeapSort.h"
using namespace  std;

int _tmain(int argc, _TCHAR* argv[])
{
    srand(int(time(NULL)));
    int A[20];
    for ( int i = 0; i < 20; i++)
    {
        A[i] = ((int)rand())%1000;
        cout << A[i] << ",";
    }
    cout << " After Heap Sort :"<<endl;
    HeapSort(A, 20);
    for ( int i = 0; i < 20; i++)
    {
        cout << A[i] << ",";
    }
    return 0;
}