大顶堆为例 —— 保证a[1]为最大值
插入push
先插在最后,然后向上维护,比自己父亲大就交换
删除pop
删除堆顶元素,先交换a[1]与a[n],然后n–,删去元素,接着从堆顶向下维护,取儿子中大的那个与自己交换,直到不存在
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define mp make_pair
#define pb push_back
#define __T int T; scanf("%d",&T);while(T--)
const ll mod=1e9+7;
int n = 0;
int a[1000];
void up(int i){
int p = i/2;
while(p!=0 && a[p]<a[i]){
swap(a[i],a[p]);
i = p;
p /= 2;
}
}
void push(int num){
a[++n] = num; //入堆
up(n); //向上维护
}
void down(int i){
int p = i*2;
if(p+1<=n && a[p+1]>a[p]) p++; //如果又儿子大,选右儿子
while(p <= n && a[i]<a[p]){
swap(a[i],a[p]);
i = p;
p *= 2;
if(p+1<=n && a[p+1]>a[p]) p++;
}
}
void pop(){
if(n==0) return;
swap(a[1],a[n--]); //把第一个元素放到最后然后删去,从堆顶向下维护
down(1);
}
int main(){
push(1);
push(3);
push(2);
cout << a[1] << " " << a[2] << " " << a[3] << endl;
pop();
cout << a[1] << " " << a[2] << endl;
return 0;
}