编程练习——二叉树（BinaryTree）

先写一个树节点类。使用Temple也可以，不使用的话，使用特定类型，用法比较局限。不过本着“今天尽量不要做出今天不必要的决定”极限编程法制。做一个特殊类型的BinaryTree，也未尝不可。

这里的TreeNode只有数据域。

 

1. template<class T>
2. class TreeNode
3. {
4. private:
5.  T data;
6.  TreeNode<T> * left;
7.  TreeNode<T> * right;
8. public:
11.  TreeNode(T& data)
12.  {
13.   this->data = data;
14.   this->left = NULL;
15.   this->right = NULL;
16.  }
18.  TreeNode(T& data,TreeNode<T>* left, TreeNode<T>* right)
19.  {
20.   this->data = data;
21.   this->left = left;
22.   this->right = right;
23.  }
25.  ~TreeNode()
26.  {
28.   this->left = NULL;
29.   this->right = NULL;
30.   cout << "node ("<<this->data<<") destory!"<<endl;
31.   // to do
32.  }
34.  T getData()
35.  {
36.   return data;
37.  }
39.  void setData(T& data)
40.  {
41.   this->data = data;
42.  }
44.  TreeNode<T> * getLeft()
45.  {
46.   return left;
47.  }
49.  void setLeft(TreeNode<T> * left)
50.  {
51.   this->left = left;
53.  }
55.  TreeNode<T> * getRight()
56.  {
57.   return right;
58.  }
60.  void setRight(TreeNode<T>* right)
61.  {
62.   this->right = right;
63.  }
66. };

下面是BinaryTree的实现。一般来说，一个BinaryTree最重要的函数就是构造和析构。基本的函数是遍历。

下面的类只实现了中序遍历，和树形打印。析构删除节点使用的是后序删除法。

1. template<class T>
2. class BinaryTree
3. {
4. protected:
5.  TreeNode<T>* head;
7.  void clearSubTree(TreeNode<T>* node)
8.  {
9.   if(node != NULL)
10.   {
11.    clearSubTree(node->getLeft());
12.    clearSubTree(node->getRight());
13.    delete node;
14.   }
15.  }
17.  void printTree(TreeNode<T>* subTree, int count)
18.  {
19.   if(subTree!=NULL)
20.   {
21.    
22.    printTree(subTree->getRight(), count+1);
23.    for(int i = 0; i < count; i++)
24.    {
25.     cout << " ";
26.    }
27.    cout <<subTree->getData()<<endl;
28.    printTree(subTree->getLeft(),count+1);
29.   }
30.  }
33.  void printInMidOrder(TreeNode<T>* subTree)
34.  {
35.   if(subTree!=NULL)
36.   {
37.    printInMidOrder(subTree->getLeft());
38.    
40.    cout <<subTree->getData()<<endl;
41.    printInMidOrder(subTree->getRight());
42.   }
43.  }
44. public:
45.  BinaryTree(T& data)
46.  {
47.   head = new TreeNode<T>(data);
48.  }
50.  BinaryTree(TreeNode<T>* head)
51.  {
52.   this->head = head;
53.  }
55.  ~BinaryTree()
56.  {
57.   clearSubTree(head);
58.  }
60.  void printTree()
61.  {
62.   printTree(this->head,0);
63.  }
65.  void printInMidOrder()
66.  {
67.   printInMidOrder(this->head);
68.  }
69. };

如果BinaryTree中还有插入删除操作就好了。但是如何插入，如何删除必须按照一定的规则进行。所以就有后面的二叉查找树、AVLTree、SplayTree、RedBlackTree等各种类型的二叉树。