Heap

heap,堆是一种完全二叉树,所以可以选择array来存储所有节点。实现heap仅需要一个array和一组heap算法(用来插入、删除元素,取极值,将某一整组数据排列成一个heap),根据元素排列方式,分为大根堆和小根堆。
heap中所有元素都遵循特别的排列规则,所以 heap不提供遍历功能,也不提供迭代器。

heap算法:C++代码实现,非STL源码。

//heap算法,底层用vector实现
#include<iostream>
#include<vector>
#include<iterator>
using namespace std;
vector<int> vec;
int max(vector<int>& vec,int i)
{
    int tmp=i-1;
    if(vec[tmp]<vec[2*i-1])
        tmp=2*i-1;
    if(vec[tmp]<vec[2*i])
        tmp=2*i;
    return tmp;
}
void swap(int& num1,int& num2)
{
    int tmp=num1;
    num1=num2;
    num2=tmp;
}
//初始化vector数据
void initVector()
{
    for(int i=0;i<10;i++)
        vec.push_back(i);
}
//对当前节点所在的只有三个节点存在的树进行排序
void sort_heap(vector<int>& vec,int k,int len)
{
    int tmp=k;
    int i=2*k+1;
    while(i<len)
    {
        if(i*2+1<len)
            sort_heap(vec,i,len);
        if(i*2+2<len)
            sort_heap(vec,i+1,len);
        if(vec[i]<vec[i+1])
            i++;
        if(vec[tmp]>vec[i])
            break;
        tmp=i;
        i=i*2+1;
    }
    if(k!=tmp)
        swap(vec[tmp],vec[k]);
    return;
}
//构建大根堆
void make_heap(vector<int>& vec,int len)
{
    for(int i=0;i<len;i++)
        sort_heap(vec,i,len);
}
//输出heap
void show_heap(vector<int>& vec)
{
    vector<int>::iterator first=vec.begin();
    vector<int>::iterator last=vec.end();
    while(first!=last)
    {
        cout<<*first<<" ";
        ++first;
    }
    cout<<endl;
}
//插入新元素后向上调整
void push_heap(int k)
{
    int tmp=vec[k];
    int i=(k+1)/2-1;
    while(i>=0)
    {
        if(vec[i]>=tmp)
            break;
        else
        {
            vec[k]=vec[i];
            k=i;
            i=(k+1)/2-1;
        }
    }
    vec[k]=tmp;
}
//删除堆顶元素并向下调整
void pop_heap(vector<int>& vec,int len)
{
    swap(vec[0],vec[len-1]);
    show_heap(vec);
    make_heap(vec,len-2);
    vec.pop_back();
}
int main()
{
    initVector();
    int len=vec.size();
    //构建大根堆
    make_heap(vec,len);
    show_heap(vec);
    //插入新元素
    vec.push_back(len++);
    push_heap(len-1);
    show_heap(vec);
    //删除堆顶元素
    pop_heap(vec,len);
    show_heap(vec);
    return 0;
}


  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值