堆的简介
Heap是一种数据结构具有以下的特点:
1)完全二叉树
2)heap中存储的值是偏序
小根堆: 父节点的值小于或等于子节点的值
大根堆: 父节点的值大于或等于子节点的值
堆的存储
一般都用数组来表示堆,i结点的父结点下标就为(i–1)/2。它的左右子结点下标分别为2 * i + 1和2 * i + 2。如第0个结点左右子结点下标分别为1和2。
注意:这里下标从0开始比较方便,因为0开始的话2x也是0,其子结点也是0。
堆的基本操作
1.down
当堆中某个结点的值变大,比如下面的根结点,则与其最小的子节点做替换,一直往下移动,直至满足堆的特点。
2.up
当堆中某个结点的值变小,比如下面的子结点,则与其最小的父节点做替换,一直往上移动,直至满足堆的特点。
堆的操作:删除小根堆堆的最小元素
堆中每次都删除第0个数据。为了便于重建堆,实际的操作是将最后一个数据的值赋给根结点,堆的元素个数-1,然后再从根结点开始进行一次从上向下down的调整。
堆的操作:小根堆插入元素
插入一个元素:新元素被加入到heap的末尾,然后up更新树以恢复堆的次序。
例题如下:
#include <iostream>
const int N = 100010;
int h[N];
int n, m, num;
using namespace std;
void down(int i)
{
int t = i;
if (2 * i <= num && h[2 * i] < h[t])
{
t = 2 * i;
}
if (2 * i + 1<= num && h[2 * i + 1] < h[t])
{
t = 2 * i + 1;
}
if (i != t)
{
swap(h[i], h[t]);
down(t);
}
}
void up (int i)
{
while(i/2 && h[i/2] > h[i])
{
swap(h[i/2], h[i]);
i = i /2;
}
}
int main()
{
scanf("%d%d", &n, &m);
num = n;
for (int i = 1; i <= n; i++)
{
scanf("%d", &h[i]);
}
for(int i = n/2; i; i --) { //创建堆时,一定要从后往前down,这样才能将后面的值往前引进
down(i);
}
while(m --)
{
printf("%d ", h[1]);
h[1] = h[num];
num --;
down(1);
}
}