把一组数放到堆(heap[])里,把堆抽象成一个完全二叉树(令堆顶元素的下标为1),若某元素的下标为x,则左边的子节点的下标为2*x,右边的子节点的下标为2*x+1。
一、up函数:
void up(int u)//如果该元素的父节点对应的元素heap[u/2]>该元素heap[u],则将heap[u]向上调整;
{
while(u/2>=1&&heap[u/2]>heap[u])
{
swap(heap[u/2],heap[u]);
u/=2;
}
}
向堆里插入一个数x:heap[++size]=x; up(size);(size为堆里的最后一个元素的下标)
每一个插入堆的元素,都需要用up函数按照大小调整该元素的位置(小的元素在大的元素的上面)
二、down函数:
void down(int u)
{
int t=u; //u*2<=size和u*2+1<=size分别保证了heap[u]有左边的子节点和右边的子节点
if(u*2<=size&&heap[u*2]<heap[t]) t=u*2;//如果heap[u*2]<heap[u],让指针t指向u*2;
if(u*2+1<=size&&heap[u*2+1]<heap[t]) t=u*2+1;//如果heap[u*2+1]<heap[u],让指针t指向u*2+1;
if(u!=t)
{
swap(heap[u],heap[t]);
down(t);//递归处理
}
}
1)求集合中的最小值:heap[1];(每一个元素的两个子节点的元素小于等于该元素)
2)删除最小值:heap[1]=heap[size]; size--; down(1);
3)删除任意一个元素:heap[k]=heap[size]; size--;
up(k); down(k);
4)修改任意一个元素:heap[k]=x;
up(k); down(k);
三、把一组数放到堆里(O(n)做法):
for(int i=n/2;i>=0;i--)
down(i);
四、堆排序(即把堆里的元素打印出来)
while(n--)
{
printf("%d ",heap[1]);
heap[1]=heap[size];
size--;
down(1);
}