Дополнить класс, включив метод подсчета числа узлов заданного бинарного дерева

/*Программа предлагает ввести значения для помещения их в дерево бинарного поиска, затем выводит результаты обхода этого дерева в префиксном, инфиксном и постфиксном порядке.*/
 
//treetest.cpp 
//Проверка класса Tree 
#include <iostream> 
#include <iomanip> 
#include "tree.h" 
using namespace std; 
int main()
{ 
//Создание объекта класса Tree<int> 
 Tree<int> intTree; 
 int intVal; 
 cout <<"Input 10 integer number: " << endl; 
//Заполнение бинарного дерева поиска 
 for (int i=0; i<10; i++) 
 { 
 cin>> intVal; 
 intTree.insertNode(intVal); 
 } 
//Вывод значений узлов бинарного дерева поиска 
//прямым, последовательным и обратным обходом 
 cout << endl << "Prefix ordered Traversal" << endl; 
 intTree.preOrderTraversal(); 
 cout << endl << "Infix ordered Traversal" << endl; 
 intTree.inOrderTraversal(); 
 cout << endl << "Postfix ordered Traversal" << endl; 
 intTree.postOrderTraversal(); 
//Создание объекта класса Tree<float> 
 Tree<float> floatTree; 
 float floatVal; 
//Установка параметров вывода с помощью манипуляторов, 
//описанных в <iomanip> 
 cout << endl << endl << endl 
 <<"Input 10 float number: " 
 << endl << setiosflags(ios::fixed | ios::showpoint) 
 << setprecision (1); 
//Вывод значений узлов бинарного дерева поиска 
//прямым, последовательным и обратным обходом 
 for(int i=0; i<10;i++) 
 { 
 cin >>floatVal; 
 floatTree.insertNode(floatVal); 
 } 
 cout << endl << "Prefix ordered Traversal" << endl; 
 floatTree.preOrderTraversal(); 
 cout << endl << "Infix ordered Traversal" << endl; 
 floatTree.inOrderTraversal(); 
 cout << endl << "Postfix ordered Traversal" << endl; 
floatTree.postOrderTraversal(); 
 cout << endl; 
 return 0; 
} 
//Tree.h 
#ifndef TREE_H 
#define TREE_H 
#include <iostream> 
#include <assert.h> 
#include "treenode.h" 
//Объявление объекта для узла бинарного дерева поиска 
//с помощью шаблона бинарного дерева поиска 
template<class NODETYPE> 
class Tree 
{ 
 public: 
 Tree(); //Конструктор 
//Помещение значения в узел дерева 
 void insertNode(const NODETYPE&); 
//Обход дерева заданным способом и печать значений в узлах 
 void preOrderTraversal() const; 
 void inOrderTraversal() const; 
 void postOrderTraversal() const; 
 private: 
//Объявление указателя на корневой узел дерева 
 TreeNode<NODETYPE> *rootPtr; 
//Помещение узла в дерево в качестве концевого узла 
 void insertNodeHelper(TreeNode<NODETYPE> **,const NODETYPE&); 
 void preOrderHelper(TreeNode<NODETYPE> *) const; 
 void inOrderHelper(TreeNode<NODETYPE> *) const; 
 void postOrderHelper(TreeNode<NODETYPE> *) const; 
}; 
//Конструктор, изначально дерево объявляется пустым 
template<class NODETYPE> 
Tree<NODETYPE>::Tree(){ rootPtr=NULL;} 
template<class NODETYPE> 
void Tree<NODETYPE>::insertNode(const NODETYPE& value) 
 { insertNodeHelper(& rootPtr, value); 
 } 
//Реализация механизма вставки нового значения в 
//бинарное дерево поиска 
template<class NODETYPE> 
void Tree<NODETYPE>::insertNodeHelper(TreeNode<NODETYPE> ** ptr, 
 const NODETYPE& value) 
{ 
 if (*ptr==NULL) 
 { 
 *ptr=new TreeNode<NODETYPE>(value); 
 assert(*ptr!=NULL); 
 } 
else 
 if(value<(*ptr)->data) 
 insertNodeHelper(& ((*ptr)->leftPtr),value); 
 else 
 if(value>(*ptr)->data) 
 insertNodeHelper(& ((*ptr)->rightPtr),value); 
 else 
 cout << value << "Dub" <<endl; 
} 
 
template<class NODETYPE> 
void Tree<NODETYPE>::preOrderTraversal() const 
 { preOrderHelper(rootPtr);} 
 
template<class NODETYPE> 
void Tree<NODETYPE>::preOrderHelper(TreeNode<NODETYPE> *ptr) const 
{ 
 if (ptr!=NULL) 
 { 
 cout <<ptr->data <<' '; 
 preOrderHelper(ptr->leftPtr); 
 preOrderHelper(ptr->rightPtr); 
 } 
} 
 
template<class NODETYPE> 
void Tree<NODETYPE>::inOrderTraversal() const 
 { inOrderHelper(rootPtr);} 
 
template<class NODETYPE> 
void Tree<NODETYPE>::inOrderHelper(TreeNode<NODETYPE> *ptr) const 
{ 
 if (ptr!=NULL) 
 { 
 inOrderHelper(ptr->leftPtr); 
 cout <<ptr->data <<' '; 
 inOrderHelper(ptr->rightPtr); 
 } 
} 
 
template<class NODETYPE> 
void Tree<NODETYPE>::postOrderTraversal() const 
 { postOrderHelper(rootPtr);} 
 
template<class NODETYPE> 
void Tree<NODETYPE>::postOrderHelper(TreeNode<NODETYPE> *ptr) const 
{ 
 if (ptr!=NULL) 
 { 
 postOrderHelper(ptr->leftPtr); 
postOrderHelper(ptr->rightPtr); 
 cout <<ptr->data <<' '; 
 } 
} 
#endif 
 
//treenode.h Определение класса TreeNode 
#ifndef TREENODE_H 
#define TREENODE_H 
//Прототип класса Tree 
 
template<class NODETYPE> 
class Tree; 
//Объявление объекта для узла бинарного дерева с помощью 
//шаблона бинарного дерева 
template<class NODETYPE> 
class TreeNode 
{ 
//Объявление класса Tree, как дружественного данному классу, 
//что обеспечит классу Tree доступ к закрытым данным класса 
//TreeNode (*leftPtr,*rightPtr, data) 
 
 friend class Tree<NODETYPE>; 
 public: 
 TreeNode(const NODETYPE&);//Конструктор 
 NODETYPE getData() const; //Возвращение данных 
 private: 
 TreeNode *leftPtr; //Указатель на левый дочерний узел 
 NODETYPE data; //Значение в узле дерева 
 TreeNode *rightPtr; //Указатель на правый дочерний узел 
}; 
//Конструктор устанавливает начальное значение узла (первым 
//будет идти корень) и нулевые указатели на левое и правое 
//поддеревья 
template<class NODETYPE> 
TreeNode<NODETYPE>::TreeNode(const NODETYPE& d) 
{ 
 data = d; 
 leftPtr=rightPtr=NULL; 
} 
//Возвращение копии значений данных 
 
template<class NODETYPE> 
NODETYPE TreeNode<NODETYPE>::getData() const 
{ return data;} 
 
#endif