堆(heap)是计算机科学中一类特殊的数据结构的统称。堆通常是一个可以被看做一棵树的数组对象。堆总是满足下列性质:
1.堆中某个节点的值总是不大于或不小于其父节点的值;
2.堆总是一棵完全二叉树。
手写堆用一维数组来储存数据,本篇以小根堆(某个节点总是大于其父节点)为例。
如果一个根节点的下标为x,则其左子节点的下表为2x,右子节点的下标为2x+1。
堆的基础操作可以用down和up两个操作来完成
1.插入一个数。
2.求集合当中的最小值。
3.删除最小值
4.删除任意一个元素。
5.修改任意一个元素。
核心操作down和up
down操作的思路是将父节点与两个子节点比较,找到子节点中最小的小于自己的子节点,与其交换位置。
void down(int u)
{
int t=u;
if(u*2<=size&&h[u*2]<h[t])
t=u*2;
if(u*2+1<=size&&h[u*2+1]<h[t])
t=u*2+1;
if(u!=t)
{
swap(h[u],h[t]);
down(t);
}
}
up操作则是如果当前节点的父节点更比自己大的话,就与父节点交换,直到这个值所在的节点比两个子节点都小。
void up(int u)
{
while(u/2&&h[u/2]>h[u])
{
swap(h[u/2],h[u]);
u/=2;
}
}
如何建立一个堆?
建堆的操作可以使每读入一个数,就执行一次插入操作,不过有一种时间复杂度更小的操作。
for(int i=n/2;i;i--)
down(i);
先读入所有的数,再从下标为n/2的节点从小到大执行一遍down操作即可建立堆
如何利用down和up执行基本操作?
1.插入操作
h[++size]=x;up(size); 读入一个数后,对这个数进行up操作。
2.求集合当中的最小值,h[1]就是最小值。
3.删除最小值,先让堆中的最后一个数覆盖h[1],再对这个位于第一个节点的值进行down操作。
h[1]=h[size];size--;down(1);
4.删除任意一个元素,先让堆中最后一个数覆盖到被删除元素的节点,我们并不知道被删除的元素和堆中最后一个数的大小关系,所以同时写上down和up两个操作,根据大小关系,两个操作仅会执行一个。
h[k]=h[size];size--;down[k];up[k];
5.修改任意一个元素。
h[k]=x;down[k];up[k];