DS二叉树--叶子数量

题目描述

计算一颗二叉树包含的叶子结点数量。

提示：叶子是指它的左右孩子为空。

建树方法采用“先序遍历+空树用0表示”的方法，即给定一颗二叉树的先序遍历的结果为AB0C00D00，其中空节点用字符‘0’表示。则该树的逻辑结构如下图。

#输入

第一行输入一个整数t，表示有t个测试数据

第二行起输入二叉树先序遍历的结果，空树用字符‘0’表示，输入t行

#输出

逐行输出每个二叉树的包含的叶子数量

#输入样例

3
AB0C00D00
AB00C00
ABC00D00E00

#输出样例

2
2
3

#参考代码

#include <iostream>
#include<string>

using namespace std;

class TreeNode{
public:
    char data;
    TreeNode *leftchild;
    TreeNode *rightchild;
    TreeNode(){
    leftchild = NULL;
    rightchild = NULL;
    }
};

class BiTree{
public:
    TreeNode *root;
    string sTree;
    int pos; 
     int leavenum;
    BiTree():root(NULL),leavenum(0){}
    
    void Create(string s){
        pos = 0;
        sTree = s;
        root = CreateTree();

    }

    TreeNode* CreateTree(){
    TreeNode* T=new TreeNode;
    char c = sTree[pos++];
    if (c == '0') T = NULL;
    else {
        T->data = c;
        T->leftchild=CreateTree();
        T->rightchild =CreateTree();
    }
    return T;

   }

    void PreOrder(TreeNode *t){
        if(t == NULL)  return;
        else if(t->leftchild==NULL&&t->rightchild==NULL){
            leavenum++;
            return;
        }
        else{           
            PreOrder(t->leftchild);
            PreOrder(t->rightchild);
        }

    }    //先序遍历实现  
};

int main(){
    int t;
    cin>>t;
    while(t--){
        string s;
        cin>>s;
        BiTree t;
        t.Create(s);
        t.PreOrder(t.root);
        cout<<t.leavenum<<endl;

    }
}