python数据结构：求二叉树节点个数

求二叉树节点个数

1.设计思想：

建立一棵二叉树，利用递归求二叉树节点个数

2.实现代码

class BinaryTreeNode(object):
    # 创建二叉树结点的函数
    def __init__(self):
        self.data = '#'
        self.LChild = None
        self.RChild = None
class BinaryTree(object):
    #创建二叉树的函数
    def CreateBinaryTree(self, Root):
        data = input('->')
        if data == '#':
            Root = None
        else:
            Root.data = data
            Root.LChild = BinaryTreeNode()
            self.CreateBinaryTree(Root.LChild)
            Root.RChild = BinaryTreeNode()
            self.CreateBinaryTree(Root.RChild)
    #求二叉树节点个数
    def leave(self,Root):
        if Root.data is '#':
            return 0
        else:
            a = 1 +self.leave(Root.LChild)+ self.leave(Root.RChild)
            return a
    # 创建一棵二叉树
    print('创建一棵二叉树\n')
    print('         4')
    print('        / \\')
    print('       5   6')
    print('      / \\  \\')
    print('     1   2  7 ')
    print('4 5 1 # # 2 # # 6 # 7 # #')
    print('请仿照上述序列，输入某一二叉树中各结点的值（#表示空结点），每输入一个值按回车换行：')
bTN = BinaryTreeNode()
bT = BinaryTree()
bT.CreateBinaryTree(bTN)
print(bT.leave(bTN))

3.测试数据及结果

测试数据：1，2，#，#，#
测试结果：2
测试数据：４,５,１，＃，＃，２，＃，＃，６，＃，７，＃，＃
测试结果：6