一个堆实现

//a strong type heap.
#define LEFT_CHILD_POS(x) (x<<1)
#define RIGHT_CHILD_POS(x) ((x<<1)+1)
#define RIGHT_SUBLING_POS(x) (x|1)
#define LEFT_SUBLING_POS(x) (x&0xFFFFFFFFE)
#define PARENT_POS(x) (x>>1)


template<typename VALUE_TYPE,typename CONTENT_TYPE,bool IS_MAX>
class Heap
{
public:
struct Elem
{
VALUE_TYPE value;
CONTENT_TYPE content;
};


Heap(int max_size)
{
assert(pow(2.0,floor(log((double)max_size)/log(2.0))) == max_size);
elem_count_ = 1;
capacity_ = max_size+1;
elem_ptr_ = new Elem[max_size+1];
}


~Heap()
{
delete []elem_ptr_;
}


void push(VALUE_TYPE value,const CONTENT_TYPE& content)
{
assert(elem_count_ < capacity_);
elem_ptr_[elem_count_].value = value;
elem_ptr_[elem_count_].content = content;
move_up(elem_count_++);
}

VALUE_TYPE pop(CONTENT_TYPE &content)
{
assert(size());
VALUE_TYPE res = elem_ptr_[1].value;
content = elem_ptr_[1].content ;
elem_ptr_[1] = elem_ptr_[--elem_count_];
move_down(1);
return res;
}

int size(void)
{
return elem_count_ - 1;
}

int capacity(void)
{
return capacity_-1;
}
private:
inline bool big_than(VALUE_TYPE v1,VALUE_TYPE v2)
{
if (IS_MAX)
{
return v1 > v2;
}else
{
return v1 < v2;
}
}


inline void swap(int p1,int p2)
{
Elem tem = elem_ptr_[p1];
elem_ptr_[p1] = elem_ptr_[p2];
elem_ptr_[p2] = tem;
}


void move_down(int i)
{
int lp = LEFT_CHILD_POS(i);
int rsp = RIGHT_SUBLING_POS(i);

if (rsp < elem_count_ 
&& big_than(elem_ptr_[rsp].value , elem_ptr_[i].value)
&& (lp >= elem_count_ || big_than(elem_ptr_[rsp].value , elem_ptr_[lp].value)))
{
swap(i,rsp);
move_down(rsp);
}else if(lp < elem_count_ && big_than(elem_ptr_[lp].value , elem_ptr_[i].value))
{
swap(i,lp);
move_down(lp);
}
}


void move_up(int i)
{


int lsp = LEFT_SUBLING_POS(i);
if (lsp && big_than(elem_ptr_[i].value , elem_ptr_[lsp].value))
{
swap(i,lsp);
move_up(lsp);
}else
{
int pp = PARENT_POS(i);
if (pp && big_than(elem_ptr_[i].value , elem_ptr_[pp].value))
{
swap(i,pp);
move_up(pp);
}
}
}


Elem *elem_ptr_;
int capacity_;
int elem_count_;

};

/*

coding is so hard, can this be more simple?

*/

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值