#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; }
堆排序-C++版(Java类似)
猜你喜欢
转载自blog.csdn.net/jirryzhang/article/details/80457784
今日推荐
周排行