二叉樹 - 最大堆

標籤：二叉樹   堆   最大堆   

maxheap.h

#include <iostream>template <typename T>class MaxHeap{public:MaxHeap(int num);MaxHeap(T Arr[], int arrsize, int totalsize);bool insert(const T&);bool del(T&);void show() const;void showonlevel() const;private:T* arr;int size, max_size;};template <typename T>MaxHeap<T>::MaxHeap(int num){arr = new T [num + 1];max_size = num;size = 0;}template <typename T>MaxHeap<T>::MaxHeap(T Arr[], int arrsize, int totalsize){arr = new T [totalsize + 1];max_size = totalsize;size = arrsize;arr = Arr;int loc, cloc;for (loc = size/2; loc >= 1; --loc){T tmp = arr[loc];cloc = loc * 2;while (cloc <= size){if (cloc < size && arr[cloc + 1] > arr[cloc])++cloc;if (tmp > arr[cloc])break;arr[cloc/2] = arr[cloc];cloc *= 2;}arr[cloc/2] = tmp;}}template <typename T>bool MaxHeap<T>::insert(const T& val){if (size == max_size)return false;int num = ++size;while (num > 1 && val > arr[num/2]){arr[num] = arr[num/2];num /= 2;}arr[num] = val;return true;}template <typename T>bool MaxHeap<T>::del(T& val){if (size == 0)return false;val = arr[1];T tmp = arr[size--];int loc = 1, cloc = 2;while (cloc <  size){if (cloc < size && arr[cloc + 1] > arr[cloc])++cloc;if (tmp > arr[cloc])break;arr[loc] = arr[cloc];loc = cloc;cloc *= 2;}arr[loc] = tmp;return true;}template <typename T>void MaxHeap<T>::show()const{for (int i = 1; i <= size; ++i)std::cout <<arr[i] <<" ";std::cout <<std::endl;}template <typename T>void MaxHeap<T>::showonlevel()const{int k = 1;for (int i = 1; i <= size; ++i){std::cout <<arr[i] <<" ";if (i == k){std::cout <<std::endl;k = k*2 + 1;}}std::cout <<std::endl;}

main.cpp

#include <iostream>#include "maxheap.h"int main(){MaxHeap<int> a(30);int c[32];for (int i = 1 ; i <= 30; i++){c[i] = rand();a.insert(c[i]);}a.showonlevel();std::cout <<std::endl;MaxHeap<int> b(c, 30,40);b.showonlevel();std::cout <<std::endl;std::cin.get();return 0;}

二叉樹 - 最大堆