【C++】二叉树的基本操作

<pre name="code" class="cpp">#include<iostream>
#include<queue>
#include<stack>
using namespace std;
struct BinaryTreeNode
{
char _value;          //节点值
BinaryTreeNode*_left; //左孩子
BinaryTreeNode*_right;//右孩子
BinaryTreeNode(const char value)
:_value(value)
,_left(NULL)
, _right(NULL)
{}
};
class BinaryTree
{
public:
BinaryTree(const char *str)
:_root(NULL)
{
_CreatBinaryTree(_root, str);
}
//先序遍历
void RootSort()
{
cout << "先序遍历：" << endl;
_RootSort(_root);
cout << endl;
}
//中序遍历
void MidSort()
{
cout << "中序遍历：" << endl;
_MidSort(_root);
cout << endl;
}
//后序遍历
void BackSort()
{
cout << "后序遍历：" << endl;
_BackSort(_root);
cout << endl;
}

//层序遍历
//思想：队列的先进先出
void LevelSort()
{
cout << "层序遍历：" << endl;
queue<BinaryTreeNode*>q;
if (_root)
{
q.push(_root);
}
while (!q.empty())
{
BinaryTreeNode*front = q.front();
q.pop();
cout << front->_value << " ";
if (front->_left)
{
q.push(front->_left);
}
if (front->_right)
{
q.push(front->_right);
}
}
cout << endl;
}
//节点个数
int Size()
{
cout << "节点个数为：";
return _Size(_root);
}
int Depth()
{
cout << "二叉树深度为：";
return _Depth(_root);
}
protected:
void _CreatBinaryTree(BinaryTreeNode*&root, const char*&str)
{
if (*str != '#'&&*str != '\0')
{
root = new BinaryTreeNode(*str);
_CreatBinaryTree(root->_left, ++str);
if (*str != '\0')
{
_CreatBinaryTree(root->_right, ++str);
}
}
}
void _RootSort(BinaryTreeNode*&root)
{
if (root)
{
cout << root->_value << " ";
_RootSort(root->_left);
_RootSort(root->_right);
}
}
void _MidSort(BinaryTreeNode*&root)
{
if (root)
{
_MidSort(root->_left);
cout << root->_value << " ";
_MidSort(root->_right);
}
}
void _BackSort(BinaryTreeNode*&root)
{
if (root)
{
_BackSort(root->_left);
_BackSort(root->_right);
cout << root->_value << " ";
}
}
int _Size(BinaryTreeNode*&root)
{
if (root==NULL)
{
return 0;
}
else
{
return 1 + _Size(root->_left) + _Size(root->_right);
}
}
int _Depth(BinaryTreeNode*&root)
{
if (root == NULL)
{
return 0;
}
if (root->_left == NULL&&root->_right == NULL)
{
return 1;
}
return 1 + (_Depth(root->_left) > _Depth(root->_right)
? _Depth(root->_left) : _Depth(root->_right));
}
private:
BinaryTreeNode*_root;
};
void Test()
{
BinaryTree bt1("123##4##5678");
bt1.RootSort();
bt1.MidSort();
bt1.BackSort();
bt1.LevelSort();
cout << bt1.Size() << endl;
cout << bt1.Depth() << endl;
}
int main()
{
Test();
system("pause");
return 0;
}