基本知识
存储:
定义一个数组,第一个节点表示根节点
对于任意一个节点x,2x为其左儿子,2x+1为其右儿子
堆是一个完全二叉树,除最后一行的元素外,其他元素均非空,最后一行从左到右依次排布
对于堆中的任意一个元素,其小于其两个儿子,所以根节点为最小值
基本操作:
down操作:对大元素进行下移
up操作:对小元素进行上移
功能:
1、插入一个数
2、取出最小值
3、删除最小值
4、修改任意一个元素
5、删除任意一个元素
题目
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int h[N], s;
void down(int u)
{
int t = u;
if (u * 2 <= s && h[u * 2] < h[t]) //与左儿子比较大小
t = u * 2;
if (u * 2 + 1 <= s && h[u * 2 + 1] < h[t]) //与右儿子比较大小
t = u * 2 + 1;
if (u != t) //如果值不同说明这个数应该存在下面一个点
{
swap(h[u], h[t]);
down(t); //再次递归处理这个数
}
}
void up(int u)
{
while (u / 2 && h[u / 2] > h[u]) //,完全二叉树一个点只有两个儿子,所以上移不需要考虑左右的问题
{
swap(h[u / 2], h[u]);
u = u / 2;
}
}
int main()
{
int n, m;
cin >> n >> m;
s = n;
for (int i = 1; i <= n; i++)
cin >> h[i];
for (int i = n / 2; i ; i--)
down(i);
while(m--)
{
cout<<h[1]<<" ";
h[1]=h[s]; //将最后一个值移动到根节点再次进行down操作,因为对尾节点操作较为容易
s--; //记得操作后堆中元素--
down(1);
}
return 0;
}