/*堆排序(大顶堆)*/
//建立一个大顶堆后,在调整成小顶堆,然后输出
#include <iostream>
#include<algorithm>
using namespace std;
//调整堆
//调整以当前节点为父节点的所有子节点和孙子节点。。。
void HeapAdjust(int *a,int i,int size)
{
int lchild=2*i; //i的左孩子节点序号
int rchild=2*i+1; //i的右孩子节点序号
int max=i; //临时变量
if(i <= size/2) //如果i是叶节点就不用进行调整
{
if(lchild<=size && a[lchild]>a[max])
{
max=lchild;
}
if(rchild<=size && a[rchild]>a[max])
{
max=rchild;
}
if(max!=i)
{
swap(a[i],a[max]);
HeapAdjust(a,max,size); //避免调整之后以max为父节点的子树不是堆
}
}
}
//自右向左,自底向上,逐个“三角”调整(一个parent和两个child)
/*
1
2 3
4 5 6 7
*/
//例如上面的树,我们先调整3,6,7这三个位置上的数,使其成为一个大顶堆,
//然后在调整2,4,5这三个数使其成为大顶堆,在调整以1为父节点的所有子节点和孙子节点。
//从而整个树都成为大顶堆了。
void BuildHeap(int *a,int size)
{
int i;
for(i=size/2;i>=1;i--) //非叶节点最大序号值为size/2
{
HeapAdjust(a,i,size);
}
}
//堆排序----把先前建立起来的大顶堆调整成小顶堆。
void HeapSort(int *a,int size)
{
int i;
BuildHeap(a,size);
for(i=size;i>=1;i--)
{
swap(a[1],a[i]); //交换堆顶和最后一个元素,即每次将剩余元素中的最大者放到最后面
HeapAdjust(a,1,i-1); //重新调整堆顶节点成为大顶堆
}
}
int main(int argc, char *argv[])
{
int a[100];
int size;
while(scanf("%d",&size)==1 && size>0)
{
int i;
cout << "排序前:";
for(i=1; i<=size; i++)
{
cin>>a[i];
}
//BuildHeap(a,size);
HeapSort(a,size);
cout << "排序后:";
for(i=1; i<=size; i++)
{
cout<<a[i]<<" ";
}
cout<<endl;
}
return 0;
}
【基础排序】堆排序
最新推荐文章于 2022-04-09 12:50:43 发布