heapsort(堆排序)

堆

(二叉)堆是一个数组，可以近似看成一个完全二叉树，有两种形式：最大堆和最小堆，树的根节点为data[1],对于任意节点i，它的父亲，左孩子，右孩子可以用以下方法得到：

int father(int i)
{
	return i/2;
}
int left(int i)
{
	return i*2; 
}
int right(int i)
{
	return i*2+1;
}

最大堆

除根节点外，都有data[father(i)]>=data[i]

堆排序步骤（最大堆）

1.MAX_HEAPIFY

MAX_HEAPIFY是用来维护最大堆的重要过程,假定根节点为left(i)和right(i)的二叉树都是最大堆，但是data[i]有可能小于其孩子，所以违背啦最大堆的性质，而MAX_HEAPIFY就是让data[i]的值“逐级下降”。

void MAX_HEAPIFY(int data[],int i)
{
	int l=left(i);
	int r=right(i);
	int large=i;
	if(data[l]>data[large]&&(l<=heapsize))
		large=l;
	if(data[r]>data[large]&&(r<=heapsize))
		large=r;
	
	if(large!=i)                         //注意交换
	{
	  int temp=0;
	  temp=data[i];
	  data[i]=data[large];
	  data[large]=temp;
	  MAX_HEAPIFY(data,large);
	}		
	  else return;			
}

2.BUILT_MAX_HEAP

首先提一下：data[father(i)+1…n]都是树的叶子节点，证明省略。

void BUILT_MAX_HEAP(int data[])
{
	for(int i=heapsize/2;i>=1;i--)
	  MAX_HEAPIFY(data,i);
}

3.堆排序

void HEAPSORT(int data[])
{
	const int len=heapsize;
	BUILT_MAX_HEAP(data);
	for(int i=len;i>1;i--)
	{
	  int temp=data[i];
	  data[i]=data[1];
	  data[1]=temp;
	  heapsize--;
	  MAX_HEAPIFY(data,1);
	}
}

完整代码如下

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
int heapsize;
int father(int i)
{
	return i/2;
}
int left(int i)
{
	return i*2; 
}
int right(int i)
{
	return i*2+1;
}
void MAX_HEAPIFY(int data[],int i)
{
	int l=left(i);
	int r=right(i);
	int large=i;
	if(data[l]>data[large]&&(l<=heapsize))
		large=l;
	if(data[r]>data[large]&&(r<=heapsize))
		large=r;
	
	if(large!=i)                         //注意交换
	{
	  int temp=0;
	  temp=data[i];
	  data[i]=data[large];
	  data[large]=temp;
	  MAX_HEAPIFY(data,large);
	}		
	  else return;			
}
void BUILT_MAX_HEAP(int data[])
{
	for(int i=heapsize/2;i>=1;i--)
	  MAX_HEAPIFY(data,i);
}
void HEAPSORT(int data[])
{
	const int len=heapsize;
	BUILT_MAX_HEAP(data);
	for(int i=len;i>1;i--)
	{
	  int temp=data[i];
	  data[i]=data[1];
	  data[1]=temp;
	  heapsize--;
	  MAX_HEAPIFY(data,1);
	}
}
int main()
{
	cin>>heapsize;
	const int len=heapsize;
	int data[100000]={0};
    for(int i=1;i<=heapsize;i++)
		cin>>data[i];
	HEAPSORT(data);
	for(int i=1;i<len;i++)
	  	cout<<data[i]<<" ";
	cout<<data[len]<<endl;
	return 0;
}