python 實現二叉樹 求二叉樹的葉子節點個數

#coding=utf-8

class Node(object):
    """節點類"""
    def __init__(self, elem=-1, lchild=None, rchild=None):
        self.elem = elem
        self.lchild = lchild
        self.rchild = rchild

class Tree(object):
    """樹類"""
    def __init__(self):
        self.root = Node()
        self.myQueue = []

    def add(self, elem):
        """為樹添加節點"""
        node = Node(elem)
        if self.root.elem == -1:  # 如果樹是空的，則對根節點賦值
            self.root = node
            self.myQueue.append(self.root)
        else:
            treeNode = self.myQueue[0]  # 此結點的子樹還沒有齊。
            if treeNode.lchild == None:
                treeNode.lchild = node
                self.myQueue.append(treeNode.lchild)
            else:
                treeNode.rchild = node
                self.myQueue.append(treeNode.rchild)
                self.myQueue.pop(0)  # 如果該結點存在右子樹，將此結點丟棄。
　　
　　

　　def leave(self,root):   #遞歸求葉子節點個數
    　　if root==None:
        　　return 0
    　　elif root.lchild ==None and root.rchild == None :
        　　return 1
    　　else:
        　　return (self.leave(root.lchild)+self.leave(root.rchild))

if __name__ == '__main__':
    """主函數"""
    elems = range(10)           #生成十個數據作為樹節點
    tree = Tree()          #新建一個樹對象
    for elem in elems:
        tree.add(elem)           #逐個添加樹的節點

　　print '\n葉子節點個數:'
　　num = tree.leave(tree.root)  
　　print num