【算法】如何手写一个堆（自用复习笔记）

堆

• 定义
  维护一个数据集合。堆是一颗完全二叉树，除了最有一层节点之外，上面所有的结点都是满的，最后一个是从左到右排布。

性质
小根堆：每个点小于等于 左右儿子（递归定义），根节点是每个树的最小值。

堆的存储

一维数组，下标从一号点开始（[1]）1号点是根节点，设x的左儿子是2x，x的右儿子是2x+1

推排序基本操作

（STL里面的堆是优先队列）

down(x)逻辑

假设堆做好了，更改一个数向下调整


因此把一个值更改变大，就往下移动

up(x)逻辑

假设堆做好了，更改一个数向上调整。

比父节点小，就更换负结点上移2

插入一个数

heap是堆，size表示当前堆大小

如果想插入一个x，就heap[++size] = x；up（size）//整个堆的最后一个新的数不断往上移动

求集合中的最小值

heap[1]

删除最小值

1.先把最后一个元素覆盖到堆顶
2.最后比较头三个，在用down
原理如果想伪装到一个tream中kill最后一个人的时候才能伪装进去。即取代堆顶
3.此时让替代变成堆顶点往下走
heap[1] = heap[size];size--;down(1);

删除任意一个元素（stl无法直接实现）

删除第k个点
if 第k 变大
1、此时down
if 第k 变小
1、此时up
if 第k 不变
1、不变
技巧
heap[k] = heap[size]；size-；down（k）up（k）//此时不论k为多少，只会在down，up中选择一个

修改任意一个元素（stl无法直接实现）

修改第k个点
if 第k 变大
1、此时down
if 第k 变小
1、此时up
if 第k 不变
1、不变
技巧
heap[k] = down（k）up（k）//此时不论k为多少，只会在down，up中选择一个

代码

输入一个长度为n的整数数列，从小到大输出前m小的数。

输入格式
第一行包含整数n和m。

第二行包含n个整数，表示整数数列。

输出格式
共一行，包含m个整数，表示整数数列中前m小的数。

数据范围
1≤m≤n≤105，
1≤数列中元素≤109
输入样例：
5 3
4 5 1 3 2
输出样例：
1 2 3

思路

//定义h[N]数组,size当前heap多少个元素;

//1、整个数组建堆（操作1-2-3实现down即可，1和3时间复杂度nlogn，2 o（1））
//简化操作1变成o（n）的时间按复杂度；
//2、从前往后输入m数，m不为0
//3、每次输出堆顶h[1]
//4、删堆顶haep[1] = h[size];size--;down(1);

函数逻辑

如果三个点最小值用t表示
如果左儿子存在并且h[u*2] < h[t] - 输出 t = u *2；
如果右儿子存在并且h[u*2+1] < h[t] - 输出 t = u *2+1；
如果u ！= t ,swap交换hu和ht；递归down （t）

#include <iostream>
#include <algorithm>

using namespace std;
const int N =100010;
int h[N],cnt ;//整个数组建堆

void down(int u){
    int t = u;
    if (u * 2 <= cnt && h[u * 2] < h[t]) t = u * 2;//表示更新在左边
    if (u * 2 + 1 <=cnt && h[u*2 + 1] < h[t]) t = u * 2 + 1;
    if (u != t) 
    {
        swap(h[u], h[t]);
        down(t);
    }
}

int main(){
    int n,m;
    cin >>n >> m;
    for (int i =1; i< n +1;i++)//从头开始插入堆
        cin >> h[i];
    cnt = n;
    
    for (int i = n/2;i >0;i--) down(i);//分左右
    while(m--){
        cout << h[1]<<' ';//输出堆顶
        h[1] = h[cnt--];//更新cnt
        down(1);//下移
    }
    puts(" ");
    return 0;
}