什么是堆
堆(heap)是计算机科学中一类特殊的数据结构的统称。堆通常是一个可以被看做一棵树的数组对象。
堆总是满足一下性质:
1.堆是一棵完全二叉树(除最后一层外,每层节点非空。最后一层从左到右排列)。
2.对于小顶堆,每个节点小于或等于左右节点(大顶堆反之)。
堆的存储方式
我们通常用一维数组来进行存储。假设我们用h[N],数组表示。
那么h[1]就是我们堆的根节点,也是堆中的最小值。而对于x节点,他的左儿子是2x节点,右儿子是2x+1节点。
堆的基本操作
down(x) 向下调整
因为要满足性质当我们的节点值变大时我们要将其进行沉底操作,直到找到合适的位置。
up(x) 向上调整
同上,节点值变小了,我们要进行上升操作,为他找到合适位置。
堆的功能实现
1.插入一个数。
h[++size] = x;up(size);
2.求集合中最小值
cout<<h[1]<<endl;
3.删除最小值
h[1] = h[size–];down(1);
4.删除任意元素
h[k] = h[size–];down(k);up(k);
5.修改任意元素
h[k] = x;down(k);up(k);
例题及代码实现
链接: link.
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6+10;
int h[N],cnt;
void add(int x)
{
h[++cnt] = x;
int t = cnt;
while(t>1&&h[t/2]>h[t])
{
swap(h[t/2],h[t]);
t/=2;
}
}
void del()
{
h[1] = h[cnt--];
int t = 1;
while(t*2<=cnt)
{
if(t*2+1<=cnt)
{
if(h[t*2]<=h[t*2+1]&&h[t*2]<=h[t])
{
swap(h[t*2],h[t]);
t = t*2;
}
else if(h[t*2+1]<=h[t*2]&&h[t*2+1]<=h[t])
{
swap(h[t*2+1],h[t]);
t = t*2+1;
}
else break;
}
else
{
if(h[t*2]<=h[t])
{
swap(h[t*2],h[t]);
t = t*2;
}
else break;
}
}
}
int main()
{
int n,m;
cin>>n>>m;
cnt = 0;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
add(x);
}
while(m--)
{
cout<<h[1]<<' ';
del();
}
return 0;
}