关闭

实用类之一-----最小堆的实现

951人阅读 评论(0) 收藏 举报

MinHeap.h文件

template<class T>
class MinHeap{
 public:
  MinHeap(T a[],int n);
  MinHeap(int ms);
  ~MinHeap();
  bool Insert(const T &x);//插入一个元素,如果空返回false,否则返回true
     bool RemoveMin(T &x);//删除最小的元素,如果空返回false,否则返回true
  void MakeEmpty();//使堆为空
  bool IsEmpty();
  bool IsFull();
  void FilterDown(const int start,const int endOfHeap);//自顶向下构造堆
  void FilterUp(const int start);//自底向上构造堆
 private:
  T *heap;
  int maxSize;
  const int defaultSize;
  int currentSize;
};
template<class T>
MinHeap<T>::MinHeap(int ms):defaultSize(100){
 maxSize = ms > defaultSize ? ms : defaultSize;
 heap = new T[maxSize];
 currentSize = 0;
}
template<class T>
MinHeap<T>::MinHeap(T a[],int n):defaultSize(100){
 maxSize = n > defaultSize ? n : defaultSize;
 heap = new T[maxSize];
 currentSize = n;
   heap = a;
 int curPos = (currentSize - 2) / 2;
 while(curPos >= 0){
   FilterDown(curPos,currentSize - 1);
   curPos--;
 }
}
template<class T>
MinHeap<T>::~MinHeap(){
 delete []heap;
}
template<class T>
void MinHeap<T>::FilterDown(const int start,const int endOfHeap){
 int i = start,j = i * 2 + 1;
 T temp = heap[i];
 while(j <= endOfHeap){
   if(j < endOfHeap && heap[j] > heap[j+1]) j++;
   if(temp < heap[j]) break;
   else{
    heap[i] = heap[j];
    i = j;
    j = 2 * i + 1;
   }
 }
 heap[i] = temp;
}
template<class T>
void MinHeap<T>::FilterUp(const int start){
 int i = start, j = ( i - 1 ) / 2;
 T temp = heap[i];
 while(i > 0){
   if(temp >= heap[j]) break;
   else{
    heap[i] = heap[j];
    i = j;
    j = (i - 1) / 2;
   }
 }
 heap[i] = temp;
}
template<class T>
bool MinHeap<T>::RemoveMin(T &x){
 if(IsEmpty()){
  cerr<<"Heap empty!"<<endl;
  return false;
 }
 x = heap[0];
 heap[0] = heap[currentSize - 1];
 currentSize--;
 FilterDown(0,currentSize-1);
 return true;
}
template<class T>
bool MinHeap<T>::Insert(const T& x){
 if(IsFull()) {
  cerr<<"Heap Full!"<<endl;
        return false;
 }
 heap[currentSize] = x;
 FilterUp(currentSize);
 currentSize++;
 return true;
}
template<class T>
bool MinHeap<T>::IsEmpty(){
  return currentSize == 0;
}
 
template<class T>
bool MinHeap<T>::IsFull(){
  return currentSize == maxSize;
}

template<class T>
void MinHeap<T>::MakeEmpty(){
 currentSize = 0
}

main.cpp测试文件:

#include<iostream>
#include"MinHeap.h"
using namespace std;

void main(){
 int a[5] = {3,2,1,4,5};
    MinHeap<int> m_heap(a,5);
 int x,i;
 for(i = 0; i < 5; i++){
   m_heap.RemoveMin(x);
   cout << x << " ";
 }
 cout << endl;

    for(i = 0; i < 5; i++){
    m_heap.Insert(a[i]);
 }

 for(i = 0; i < 5; i++){
   m_heap.RemoveMin(x);
   cout << x << " ";
 }
 cout << endl;
}

0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:280258次
    • 积分:3892
    • 等级:
    • 排名:第8087名
    • 原创:108篇
    • 转载:25篇
    • 译文:0篇
    • 评论:69条
    最新评论
    文章收藏