1 //////////////////////////////////////////////////////////////////////////
 2 //code by hzs
 3 //email: huangzhesi@gmail.com
 4 //Last modified: 2014-5-18 21:05
 5 //////////////////////////////////////////////////////////////////////////
 6 
 7 #ifndef _TREE_NODE_ALLOC_H
 8 #define _TREE_NODE_ALLOC_H
 9 
10 #include <boost/pool/pool_alloc.hpp>
11 
12 //树节点内存管理:仿SGI-STL-(rb-tree)
13 
14 //arg1: 树节点的值
15 //arg2: 树节点
16 template<typename Value, typename tree_node>
17 class TreeNodeAlloc {
18 public:
19     typedef Value value_type;
20     typedef tree_node* link_type;
21 
22     typedef boost::fast_pool_allocator<tree_node> Tree_Node_Alloc;    //使用boost::fast_pool_allocator
23     Tree_Node_Alloc tree_node_alloc_;
24 
25     link_type get_node() { return Tree_Node_Alloc::allocate(); }    //分配内存空间
26     void put_node(link_type p) { Tree_Node_Alloc::deallocate(p); }    //回收内存空间
27 
28     link_type create_node(const value_type& x) {
29         link_type tmp = get_node();
30         tree_node_alloc_.construct(tmp, tree_node());    //构造（若tree_node中均为POD则无需此步）
31         tmp->value_field = x;    //给值域赋值
32 
33         return tmp;
34     }
35 
36     void destroy_node(link_type p) {
37         tree_node_alloc_.destroy(p);    //析构（无构造则无析构）
38         put_node(p);
39     }
40 };
41 
42 #endif