堆对象的创建与实现的核心思想就是上调(adjustup)与下调(adjustdown)的算法思想,上调用于创建堆时,从第一个非叶子节点开始向根节点根据需求调整为大堆或者小堆
下调如图示:
当我们进行插入时,会影响堆的结构,这时我们用尾插,然后上调如图示:
接下来就可以创建堆类,代码如下仅供参考:
#include<iostream> #include<vector> template <class T> struct CompMax { bool operator()(const T& a, const T& b) { return a > b; } }; template <class T> struct CompMin { bool operator()(const T& a,const T& b) { return a < b; } }; template <class T,class Com=CompMax<T> >//仿函数做模板参数,可根据需求修改比较方法 class Heap { public: Heap(const T* arr, size_t size, Com _comp) :comp(_comp) { _List.resize(size); int index = 0; for (index = 0; index < size; ++index) { _List[index] = arr[index]; } index = (_List.size()- 2) / 2; while (index>=0) _adjustdown(index--); } void Push(const T &x) { _List.push_back(x); size_t index = _List.size() - 1; _adjustup(index); } void Pop() { _List.pop_back(); } T& Top() { return _List[0]; } protected: void _adjustup(size_t index) { size_t child = index; size_t parent = (child - 1) / 2; while (child) { if (child % 2) { if (child + 1<_List.size()) child =comp(_List[child],_List[child+1]) ? child : child + 1; } else { child = comp(_List[child] ,_List[child - 1]) ? child : child - 1; } if (!comp(_List[child],_List[parent])) { std::swap(_List[parent], _List[child]); } child = parent; parent = (parent - 1) / 2; } } void _adjustdown(size_t index) { size_t parent = index; size_t child = parent * 2 + 1; while (child < _List.size()) { if (child + 1 < _List.size()) child = comp(_List[child] , _List[child + 1]) ? child : child + 1; if (!comp(_List[parent], _List[child])) { std::swap(_List[child], _List[parent]); parent = child; child = (parent + 1) * 2; } else break; } } protected: vector<T> _List; Com comp; };
如有不足希望指正,如有问题也希望提出,谢谢-3-。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。