二叉树的创建、添加节点、先序、中序和后序遍历

class Node(object):
    """结点类"""
    def __init__(self, item):
        self.item = item
        self.lchild = None
        self.rchild = None


class BinaryTree(object):
    """二叉树"""
    def __init__(self, node=None):
        self.root = node

    def add(self, item):
        """
        广度优先遍历方式添加结点
        :param item:
        :return:
        """
        if self.root is None:# 空二叉树
            self.root = Node(item)# 添加结点作为根结点
        else:
            queue = []
            queue.append(self.root)

            while len(queue) > 0:# while 1或True也是可以的
                node = queue.pop(0)#从队列中取出一个结点
                if not node.lchild:#判断该结点左孩子是否为空
                    node.lchild = Node(item)
                    return
                else:
                    queue.append(node.lchild)
                if not node.rchild:#判断该结点右孩子是否为空
                    node.rchild = Node(item)# 为右结点添加新的数据
                    return
                else:
                    queue.append(node.rchild)

    def breadh_travel(self):
        """广度优先遍历"""
        if self.root is None:# 判断根结点是否为空
            return
        queue = []
        queue.append(self.root)
        while len(queue)>0:
            node = queue.pop(0)# 从队列中取出一个结点
            print(node.item, end=" ")#打印结点
            if node.lchild:
                queue.append(node.lchild)# 判断它的左孩子是否为空,不为空表示可添加
            if node.rchild:# 判断它的右孩子是否为空
                queue.append(node.rchild)

    def preorder_travel(self, root):#root也可能为子树的根
        """先序 根 左 右"""
        if root:
            print(root.item, end=" ")#打印根结点
            self.preorder_travel(root.lchild)#遍历它的左孩子
            self.preorder_travel(root.rchild)#然后再遍历右孩子

    def inorder_travel(self, root):
        """中序 左 根  右"""
        if root:
            self.inorder_travel(root.lchild)
            print(root.item, end=" ")
            self.inorder_travel(root.rchild)

    def postorder_travel(self, root):
        """后序 左 右 根"""
        if root:
            self.postorder_travel(root.lchild)
            self.postorder_travel(root.rchild)
            print(root.item, end=" ")

if __name__ == '__main__':
    tree = BinaryTree()
    tree.add(0)
    tree.add(1)
    tree.add(2)
    tree.add(3)
    tree.add(4)
    tree.add(5)
    tree.add(6)
    tree.add(7)
    tree.add(8)
    tree.add(9)
    tree.breadh_travel()
    print("")
    tree.preorder_travel(tree.root)
    print("")
    tree.inorder_travel(tree.root)
    print("")
    tree.postorder_travel(tree.root)
    print("")