Heap是一种基于完全二叉树的数据结构,形式上是保证每个根节点的数值都大于(小于)他的两个孩子节点,由于是基于完全二叉树的数据结构,我们可以利用数组来实现
相关操作代码:
const int maxn=100;
int n;//结点数量
int heap[maxn];//因为堆基于完全二叉树,所以我们用数组表示
//对heap数组在[low,high]范围内进行向下调整
//其中low为欲调整结点的数组下标,high一般为堆的最后一个元素的下标
void downAdjust(int low,int high)//向下调整,O(logN)
{
int i=low,j=i*2;//i为欲调整结点,j为其左孩子
while(j<=high)//存在孩子结点
{
if(j+1<=high&&heap[j+1]>heap[j])//如果右孩子存在且右孩子的值大于左孩子
j=j+1;//让j存储右孩子下标
if(heap[j]>heap[i])//孩子中最大的那个比要调整的结点i大
{
swap(heap[j],heap[i]);//交换二者
i=j;//始终保持i为需要调整的结点,j为i的孩子
j=i*2;
}
else break;//孩子结点权值均比欲调整结点i要小,调整结束
}
}
//因为叶节点数量是n/2,所以1~n/2都是非叶节点
void createHeap()//建堆,O(N)
{
for(int i=n/2;i>=1;i--)//注意遍历次序
downAdjust(i,n);
}
void upAdjust(int low,int high)
{
int i=high,j=i/2;//i为欲调整结点,j为其父亲
while(j>=low)
{
if(heap[j]<heap[i])//父亲权值小于欲调整结点i的权值
{
swap(heap[j],heap[i]);
i=j;//保持i为欲调整结点,j为i的父亲
j=i/2;
}
else break;
}
}
//插入是向上调整,删除是向下调整
void insert(int x)
{
heap[++n]=x;//让元素个数+1,然后把数组末位赋值为x
upAdjust(1,n);//向上调整新加入的结点n
}
void deleteTop()//删除堆顶元素,O(logN)
{
heap[1]=heap[n--];//用最后一个元素覆盖堆顶元素,并让元素个数-1
downAdjust(1,n);//向下调整堆顶元素
}
//堆排序的原理是,整个堆的根节点(树根)的值始终是未排序列中最大的
//我们每次把堆顶元素交换到未排序列的最后位置,经过n-1趟排序,整个堆就会有序
void heapSort()
{
createHeap();//建堆
for(int i=n;i>1;i++)//倒着枚举,i的大小是未排序范围,i>1是因为单个元素必然有序
{
swap(heap[i],heap[1]);//交换heap[i]和堆顶
downAdjust(1,i-1);//调整堆顶
}
}