数据结构--堆

堆,与其说是一种数据结构,不如说是一种排序算法。用堆算法,可以轻松得到一组数据中最大的或最小的元素。

其结构就是完全二叉树的顺序存储方式。即在一个数组中存储一颗完全二叉树。

通常,堆分为"大根堆"和"小根堆",前者的树顶元素是数组中最大的一个,后者是最小的一个。

常用操作:

入堆:元素被加入到堆中,以小根堆为例,新元素首先被放到队列最尾端,然后何其父节点比较,如果小于其父节点,则交换两者。以此类推,一直追溯到树顶元素,或者在半途,它不再小于它的父节点。操作完成。

出堆:出堆操作,就是得到当前队列中最大或最小的元素。对于一个合法的堆结构,其首元素就是所要找的。关键是这个元素取出后,其他元素要填补其移走的空白。和入堆操作相反,出堆操作是从树顶至叶节点的操作。

将一个队列构建成堆:这个操作相对复杂一些,每次都从堆底向堆顶找到其最大或最小的元素,然后分别对起两个子节点作此操作。

 

  1 #include "Array.h"
  2 #include <utility>
  3 #include <algorithm>
  4 template <typename T>
  5 class heap
  6 {
  7 public:
  8     heap(){}
  9     
 10     void push_heap(const T& data)
 11     //入堆
 12     {
 13         if(m_array.empty())
 14         {
 15             m_array.push_back(data);
 16         }
 17         else
 18         {
 19             heapify(data);
 20         }
 21     }
 22 
 23     T pop_heap()
 24     //出堆
 25     {
 26         std::swap(m_array[0], m_array[m_array.size()-1]);
 27         int i = 0;
 28         if(m_array.size()%2 == 1 && m_array.size() > 2 && m_array[m_array.size()-2< m_array[0])
 29         {
 30             std::swap(m_array[0], m_array[m_array.size()-2]);
 31         }
 32         while(i*2+2 < m_array.size()-1)
 33         {
 34             if(m_array[i*2+1< m_array[i*2+2])
 35             {
 36                 std::swap(m_array[i], m_array[i*2+1]);
 37                 i = i*2+1;
 38             }
 39             else if(i*2+2 < m_array.size()-1)
 40             {
 41                 std::swap(m_array[i], m_array[i*2+2]);
 42                 i = i*2+2;
 43             }
 44             
 45 
 46         }
 47     
 48         T tmp = m_array[m_array.size()-1];
 49         m_array.remove(m_array.end()-1);
 50         
 51         return tmp;
 52     }
 53 
 54     template <typename IT>
 55     void insert(IT _First, IT _Last)
 56     //将数组插入堆,并保持堆的合法性
 57     {
 58         for(;_First != _Last; )
 59         {
 60             push_heap(*_First++);
 61         }
 62     }
 63 
 64     template <typename IT>
 65     static void make_heap(IT _First, IT _Last)
 66     //把现有队列创建为堆结构
 67     {
 68         size_t dist = (_Last - _First-1/2;
 69         for(int i = 0; i < dist; i ++)
 70             _make_heap(_First, _Last,i);
 71     }
 72 
 73 private:
 74     void heapify(const T& data)
 75     {
 76         size_t off = m_array.size();
 77         m_array.push_back(data);
 78 
 79         int i = (off-1)/2;
 80 
 81         do
 82         {
 83             if(m_array[i] > m_array[off])
 84             {
 85                 std::swap(m_array[i], m_array[off]);
 86             }
 87             else
 88             {
 89                 break;
 90             }
 91             off = (off-1)/2;
 92             i = (off-1)/2;
 93         }while(off > 0);
 94     }
 95 
 96     template <typename IT>
 97     static void _make_heap(IT _First, IT _Last, int len)
 98     {
 99         int n = _Last - _First;
100         int nLength = (_Last - _First)/2 -1;
101 
102         if(nLength <= 1)
103         {
104             return;
105         }
106 
107         if(nLength%2 == 0)
108         //当堆的长度为偶数时,即完全二叉树上的最后一个叶子节点是其父节点的左孩子,且没有兄弟节点。
109         //则对二叉树中最后一棵子树特殊处理,下面的循环中不再理会它
110         {
111             if(*(_First+nLength) > *(_First+(nLength*2+1)))
112             {
113                 std::swap(*(_First+nLength), *(_First+(nLength*2+1)));
114                 
115             }
116             nLength --;
117         }
118 
119         for(; nLength >= len; nLength --)    
120         {
121             T a = *(_First+(nLength*2+1));
122             T b = *(_First+(nLength*2+2));
123             if(*(_First+nLength) < min(*(_First+(nLength*2+1)), *(_First+(nLength*2+2))))
124             {
125                 continue;
126             }
127             else if(*(_First+(nLength*2+1)) < *(_First+(nLength*2+2)))
128             {
129                 std::swap(*(_First+nLength), *(_First+(nLength*2+1)));
130             }
131             else
132             {
133                 std::swap(*(_First+nLength), *(_First+(nLength*2+2)));
134             }
135         }
136     }
137 
138     Array<T> m_array;
139 };
140 
posted @ 2008-10-11 22:32  刺儿头  阅读(836)  评论(1编辑  收藏  举报