堆排序-C++版(Java类似)

#include <algorithm>
#include <iostream>
using namespace std;
//小顶堆向下调整,按升序排序或最大的k个值时建大顶堆,按降序排序或计算最大的k个值时建小顶堆。
void MinHeapFixDown(int* a, int i, int n){
    int j = 2*i+1;
    int temp = a[i];
    while(j<n)
    {
        if(j+1<n&&a[j]>a[j+1])//用子节点最小的和待定节点比较。大顶堆向下调整时将">"换成"<",即用子节点最大的和待定节点比较。
            ++j;
        if(temp<a[j])//当子节点最小的都比待定节点大,则满足小顶堆的条件,当前位置即为待定节点的最终位置。大顶堆向下调整时将"<"换成">"即可。
            break;
        else
        {
            a[i]=a[j];
            i=j;
            j=2*i+1;
        }
    }
    a[i]=temp;
}

//大顶堆向下调整
void MaxHeapFixDown(int* a, int i, int n){
    int j = 2*i+1;
    int temp = a[i];
    while(j<n)
    {
        if(j+1<n&&a[j]<a[j+1])//用子节点最大的和待定节点比较。
            ++j;
        if(temp>a[j])//当子节点最大的都比待定节点小,则满足大顶堆的条件,当前位置即为待定节点的最终位置。
            break;
        else
        {
            a[i]=a[j];
            i=j;
            j=2*i+1;
        }
    }
    a[i]=temp;
}
//堆排序
void HeapSort(int *a, int n){
    for(int i= n/2-1;i>=0;i--)//从最后一个非叶节点开始,建立小顶堆
//        MinHeapFixDown(a,i,n);//建小顶堆
        MaxHeapFixDown(a,i,n);//建大顶堆
    for(int i=n-1;i>=1;i--)
    {
        swap(a[i],a[0]);
//        MinHeapFixDown(a,0,i);//降序排序
        MaxHeapFixDown(a,0,i);//升序排序
    }
}

void main()
{
    int a[]={4,5,1,6,2,7,3,8};
    HeapSort(a,8);
    for(int i=0;i<8;i++)
        cout<<a[i]<<endl;
}

猜你喜欢

转载自blog.csdn.net/jirryzhang/article/details/80457784