heap 堆

懒,不想说堆是啥了,就说说咋实现一个大顶堆吧
用一个数组保存一个二叉堆,就像二叉树一样,一个节点(下标为 i n d e x index index),其儿子节点分别为 i n d e x × 2 和 i n d e x × 2 + 1 index\times2和index\times2+1 index×2index×2+1
如果是一个大顶堆,就满足每个节点的值都比它的两个儿子的值大,显然堆顶就是最大值,对应的就是 h e a p [ 1 ] heap[1] heap[1]

入堆操作:
将一个元素,放入堆尾部,(即 h e a p [ + + i n d e x ] = v a l heap[++index]=val heap[++index]=val),然后从堆的底部开始向上(父节点)更新整个堆,因为是大顶堆,因此如果节点的值比父节点大,就交换,直到不比父节点大或者到达堆顶为止。

出堆操作:
出堆肯定是堆顶元素出来,将堆顶元素和最后一个元素交换,然后记录数量的下标 i n d e x − − index-- index就删掉了,删除后要更新堆,这次是自上而下更新,取左右两个儿子的最大值,和当前值比较,如果当前值小,就对应的交换值,注意判断下标是否合法,直到当期的值大于子节点或者到达了堆的底部为止。

/*
heap
大顶堆
*/
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int heap[N];
int ind;
void up(int n){
    while (n > 1){// 堆顶警告
        if (heap[n] > heap[n / 2]){// 当前值大与父节点就换
            swap(heap[n], heap[n / 2]);
            n = n / 2;
        }
        else break;
    }
}
void push(int val){
    // 插入采用向上交换
    heap[++ind] = val;
    int i = ind;
    up(ind);
}
void down(int n){
    while (n <= ind){// 到达堆底部
        int Son = n * 2;// 左儿子
        if (Son > ind)break;// 如果没有左儿子就更没有右儿子
        if (Son + 1 <= ind && heap[Son] < heap[Son + 1]){
            Son++;// 两个儿子都有取一个大的
        }
        if (heap[n] < heap[Son]){// 当前值小就交换
            swap(heap[n], heap[Son]);
            n = Son;
        }
        else break;
    }
}
void pop(){
    swap(heap[ind], heap[1]);
    ind--;
    down(1);
}
int top(){
    return heap[1];
}
int main()
{
    ind = 0;
    push(2);
    push(5);
    push(7);
    push(10);
    push(8);
    push(9);
    while (ind){
        printf("%d ", top());
        pop();
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值