手寫堆(heap)

手寫堆（heap）

有點閒，突然就想手寫一個堆（heap）。手寫一個就可以隨心所欲地切換自己想抽出的數據是最小還是最大了(*￣▽￣)y
二叉堆還有一個特性就是每個節點的父節點可以保證比它大，若是取最小值則爲比它小。

直接上代碼

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int MAX_N = 1e5+10;
int heap[MAX_N], sz = 0;
void push(int x) {  //用作插入數值
    //自己節點的編號
    int i = sz++;

    while(i > 0) {
        // 父親節點的編號
        int p = (i - 1)/2;
        // 如果已經沒有大小顛倒則退出
        if(heap[p] <= x)
            break;
        // 把父親節點的數值放下來，而把自己提上去
        heap[i] = heap[p];
        i = p;
    }
    heap[i] = x;
}
int pop() {     //用作提取並刪除最小值
    // 最小值
    int ret = heap[0];
    // 要提到根的數值
    int x = heap[--sz];
    heap[sz] = 0;   //把提上去的數值清零
    // 從根開始向下交換
    int i = 0;
    while(i*2 + 1 < sz) {
        //比較兒子的值
        int a = i*2 + 1, b = i*2 + 2;
        if(b < sz && heap[b] < heap[a])
            a = b;
        //如果沒有大小顛倒則退出
        if(heap[a] >= x)
            break;
        // 把兒子的數值提上來
        heap[i] = heap[a];
        i = a;
    }
    heap[i] = x;
    return ret;
}
int main() {
    int temp, n;
    scanf("%d", &n);
    for(int i = 0; i < n; i++) {
        scanf("%d", &temp);
        push(temp);
    }
    printf("%d\n", pop());
    for(int i = 0; i < n-1; i++) {
        printf("%d ", heap[i]);
    }
    printf("%d\n", heap[n-1]);
    return 0;
}