二叉查找树

1. #include "stdafx.h"
2. #include <iostream>
3. using namespace std;
5. template<typename object>
6. class less
7. {
8. public:
9.     bool operator()(object lhs,object rhs)
10.     {
11.         return lhs < rhs;
12.     }
13. };
15. template<typename object,typename comparator = less<object> >
16. class BinarySearchTree
17. {
18.     struct BinaryNode
19.     {
20.         object element;
21.         BinaryNode* left;
22.         BinaryNode* right;
23.     };
25. public:
26.     BinarySearchTree()
27.     {
28.         
29.     }
31.     ~BinarySearchTree()
32.     {
33.         //todo:
34.     }
36.     BinaryNode* findMin() const;
37.     BinaryNode* findMax() const;
38.     bool contains(const object& x) const;
39.     void printTree(ostream& out = cout);
41. private:
42.     
43.     BinaryNode* root;
44.     comparator isLessThan;
46.     bool contains(const object& x,BinaryNode* t) const; 
47.     BinaryNode* findMin(BinaryNode* t) const
48.     {
49.         if (t == NULL)
50.         {
51.             return NULL;
52.         }
54.         if (t->left == NULL)
55.         {
56.             return t;
57.         }
58.         
59.         return findMin(t->left);
60.     }
62.     BinaryNode* findMax(BinaryNode* t) const
63.     {
64.         if (t == NULL)
65.         {
66.             return NULL;
67.         }
69.         while(t->right != NULL)
70.         {
71.             t = t->right;
72.         }
74.         return t;
75.     }
76.     
77.     void insert(const object& x,BinaryNode* t)
78.     {
80.         if (t == NULL)
81.         {
82.             t = new BinaryNode;
83.             t->element = x;
84.             t->left = NULL;
85.             t->right = NULL;
86.         }
87.         else if (isLessThan(x,t->element))
88.         {
89.             insert(x,t->left);
90.         }
91.         else if (isLessThan(t->element, x))
92.         {
93.             insert(x,t->right);
94.         }
95.     }
97.     void remove(const object& x,BinaryNode* t)
98.     {
99.         if (t == NULL)
100.             return;
102.         if (isLessThan(x,t->element))
103.         {
104.             remove(x,t->left);
105.         }
106.         else if (isLessThan(t->element,x))
107.         {
108.             remove(x,t->element);
109.         }
110.         else if (t->left != NULL && t->right != NULL)
111.         {
112.             t->element = findMin(t->right)->element;
113.             remove(t->element, t->right);
114.         }
115.         else
116.         {
117.             BinaryNode* p = t;
118.             t = (t->left !=NULL) ? t->left : t->right;
119.             delete p;
120.         }
121.     }
123.     void printTree(BinaryNode* t,ostream& out)
124.     {
125.         if (t != NULL)
126.         {
127.             printTree(t->left, out);
128.             out << t->element << endl;
129.             printTree(t->right, out);
130.         }
131.     }
133.     void height(BinaryNode* t)
134.     {
135.         if (t == NULL)
136.             return 0;
137.         
138.         if (t->left != NULL || t->right != NULL)
139.             return 1 + max(height(t->left),height(t->right));
140.     }
141. };
143. template<typename object,typename comparator>
144. bool BinarySearchTree<object,comparator>::contains(const object& x,BinaryNode* t) const
145. {
146.     if (t == NULL)
147.     {
148.         return false;
149.     }
150.     
151.     if (isLessThan(x,t->element))
152.     {
153.         return contains(x,t->left);
154.     }
155.     else if(isLessThan(t->element,x))
156.     {
157.         return contains(x,t->right);
158.     }
159.     else
160.     {
161.         return true;
162.     }
164.     return false;
165. }
167. template<typename object, typename comparator>
168. bool BinarySearchTree<object, comparator>::contains(const object& x) const
169. {
170.     return contains(x,t);
171. }
173. template<typename object, typename comparator>
174. void BinarySearchTree<object, comparator>::printTree(ostream& out)
175. {
176.     if (t == NULL)
177.     {
178.         out << "empty tree" << endl;
179.         return;
180.     }
181.     printTree(t,out);
182. }

未完待续。