基礎數據結構-二叉樹-計算二叉樹的葉子結點數

計算一顆二叉樹包含的葉子結點數量。
提示：葉子是指它的左右孩子為空。

建樹方法采用“先序遍歷+空樹用0表示”的方法，即給定一顆二叉樹的先序遍歷的結果為AB0C00D00，其中空節點用字符‘0’表示。則該樹的邏輯結構如下圖。
輸入

第一行輸入一個整數t，表示有t個測試數據

第二行起輸入二叉樹先序遍歷的結果，空樹用字符‘0’表示，輸入t行

輸出

逐行輸出每個二叉樹的包含的葉子數量

樣例輸入

3
AB0C00D00
AB00C00
ABC00D00E00
樣例輸出

2
2
3

 

#include<iostream>
#include<string>
using namespace std;

class BiTreeNode
{
public:
    char data;        //結點數據
    BiTreeNode *LeftChild;    //左子樹指針
    BiTreeNode *RightChild;        //右子樹指針
    BiTreeNode():LeftChild(NULL),RightChild(NULL){}
    ~BiTreeNode(){}
};

class BiTree
{
private:
    BiTreeNode *Root;    //根結點指針
    int pos;
    string strTree;
    BiTreeNode *CreateBiTree();
    void PreOrder(BiTreeNode *t);
    void InOrder(BiTreeNode *t);
    void PostOrder(BiTreeNode *t);
public:
    BiTree(){};
    ~BiTree(){};
    void CreateTree(string TreeArray);
    void PreOrder();
    void InOrder();
    void PostOrder();
};

//構造二叉樹，利用先序遍歷結果建樹
void BiTree::CreateTree(string TreeArray)    //公有函數，對外接口
{
    pos = 0;
    strTree.assign(TreeArray);
    Root = CreateBiTree();
}

BiTreeNode *BiTree::CreateBiTree()    //遞歸建樹，私有函數，類內實現
{
    BiTreeNode *T;
    char ch;
    ch=strTree[pos++];
    if(ch=='0')
        T = NULL;
    else
    {
        T = new BiTreeNode();
        T->data = ch;        //生成根結點
        T->LeftChild = CreateBiTree();    //構造左子樹
        T->RightChild = CreateBiTree();    //構造右子樹
    }
    return T;
}

//定義先序遍歷函數
void BiTree::PreOrder()    //公有函數，對外接口
{
    PreOrder(Root);
}

void BiTree::PreOrder(BiTreeNode *t)    //私有函數，類內實現
{
    if(t!=NULL)
    {
        cout << t->data;    //輸出當前結點t的數據，表示t已經訪問
        PreOrder(t->LeftChild);    //先序遍歷t的左孩子
        PreOrder(t->RightChild);    //先序遍歷t的右孩子
    }
}

//定義中序遍歷函數
void BiTree::InOrder()    //公有函數，對外接口
{
    InOrder(Root);
}

void BiTree::InOrder(BiTreeNode *t)    //私有函數，類內實現
{
    if(t)
    {
        InOrder(t->LeftChild);    //中序遍歷t的左孩子
        cout << t->data;        //輸出當前結點t的數據，表示t已經訪問
        InOrder(t->RightChild);    //中序遍歷t的右孩子
    }
}

//后序遍歷函數
void BiTree::PostOrder()    //公有函數，對外接口
{
    PostOrder(Root);
}

void BiTree::PostOrder(BiTreeNode *t)    //私有函數，類內實現
{
    if(t)
    {
        PostOrder(t->LeftChild);    //后序遍歷t的左孩子
        PostOrder(t->RightChild);    //后序遍歷t的右孩子
        cout << t->data;        //輸出當前結點t的數據，表示t已經訪問
    }
}

int main(void)
{
    int t;
    BiTree T;
    string str;
    cin >> t;
    for(int i=0;i<t;i++)
    {
        cin >> str;
        T.CreateTree(str);
        T.PreOrder();
        cout << endl;
        T.InOrder();
        cout << endl;
        T.PostOrder();
        cout << endl;
    }
    return 0;
}