用数组手搓一个小顶堆

堆默认从数组下标为1开始存储。

const int N=201000;
int heap[N];
int len;

插入操作:

将元素插入到堆的末尾位置向上调整。

void up(int k){
  while(k>1&&heap[k/2]>heap[k]){
    swap(heap[k],heap[k/2]);
    k/=2;
  }
}
//len为当前存在元素长度
void Insert(int x){
  heap[++len]=x;
  up(len);
}

弹出堆顶元素:

将堆顶元素和堆中最后一个元素交换位置,将堆的长度减一再将新的堆顶元素向下调整。

void down(int k){
  while(k+k<=len){
    int j=k+k;
    if(j+1<=len&&heap[j+1]<heap[j])j++;
    if(heap[k]<=heap[j])break;
    swap(heap[j],heap[k]);
    k=j;
  }
}

void pop(){
  swap(heap[1],heap[len]);
  len--;
  down(1);
}

删除其中任意一个元素:

将该元素和队尾元素互换,如果队尾元素比该元素小尝试向上调整,如果队尾元素比该元素大尝试向下调整。

void Delete(int p){
  if(p==len){
    len--;
    return ;
  }
  int x=heap[p],y=heap[len];
  swap(heap[p],heap[len]);
  len--;
  if(y<x){
    up(p);
  }else {
    down(p);
  }
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值