二叉树的基本操作

二叉树的基本操作： 包括创建二叉树，遍历二叉树(先序、中序和后序)，求二叉树深度，求二叉树结点数，求叶子结点数。

其中创建二叉树是先序创建的，即输入的时候要按二叉树先序输入。

废话不说直接看代码

#include <iostream>
using namespace std;

struct BiTNode
{
    int data;
    BiTNode *lchild, *rchild;
};

//先序建立二叉树
void CreateBitTree(BiTNode *&root)
{
    int value;
    cin >> value;
    if (value == -1)
        root = NULL;
    else
    {
        root = new BiTNode;
        root->data = value;
        CreateBitTree(root->lchild);
        CreateBitTree(root->rchild);
    }
}

//先序遍历
void PreorderTraversal(BiTNode *root)
{
    if (root)
    {
        cout << root->data << ' ';
        PreorderTraversal(root->lchild);
        PreorderTraversal(root->rchild);
    }
}

//中序遍历
void InorderTraversal(BiTNode *root)
{
    if (root)
    {
        InorderTraversal(root->lchild);
        cout << root->data << ' ';
        InorderTraversal(root->rchild);
    }
}

//后序遍历
void PostorderTraversal(BiTNode *root)
{
    if (root)
    {
        PostorderTraversal(root->lchild);
        PostorderTraversal(root->rchild);
        cout << root->data << ' ';
    }
}

int GetDepth(BiTNode *root)
{
    if (root == NULL)
        return 0;
    else
    {
        int ldep = GetDepth(root->lchild);
        int rdep = GetDepth(root->rchild);
        return ldep >= rdep ? ldep + 1 : rdep + 1;
    }
}

int GetNodeCount(BiTNode *root)
{
    if (root == NULL)
        return 0;
    return GetNodeCount(root->lchild) + GetNodeCount(root->rchild) + 1;
}

int GetLeafCount(BiTNode *root)
{
    if (root == NULL)
        return 0;
    else if (root->lchild == NULL && root->rchild == NULL)
        return 1;
    else
        return GetLeafCount(root->lchild) + GetLeafCount(root->rchild);
}

int main()
{
    BiTNode *root = NULL;
    CreateBitTree(root);
    cout << "先序遍历：" << endl;
    PreorderTraversal(root);
    cout << "中序遍历：" << endl;
    InorderTraversal(root);
    cout << "后序遍历：" << endl;
    PostorderTraversal(root);
    cout << "二叉树深度：" << GetDepth(root) << endl;
    cout << "结点个数：" << GetNodeCount(root) << endl;
    cout << "叶子个数：" << GetLeafCount(root) << endl;
    system("pause");
}

输出结果：

创建的是如下的一颗二叉树：

                1
        2               3
    4      -1     -1       -1
-1     -1