二叉树遍历

#定义二叉树的节点
class Node(object):
    def __init__(self,elem):
        """
        param: self.elem 是节点的数据域
                self.lchild 是节点的左孩子
                self.rchild 是节点的右孩子
        """
        self.elem = elem
        self.lchild = None
        self.rchild = None

class Tree(object):
    def __init__(self):
        self.root = None
    #添加节点
    def add(self,elem):
        """
        param: elem 是传进来的数据，我们要实例化一个节点接收它，
                queue: 创建一个队列来接收和弹出节点       
        """
        #创建节点
        node = Node(elem)    
        if self.root == None:
            """如果根节点是None,则表示一颗空树，直接把该节点赋给root节点"""
            self.root = node
            return
        queue = []
        queue.append(self.root)
        while queue:
            """队列的弹出要加0，与栈相仿"""
            curNode = queue.pop(0)
            if curNode.lchild == None:
                curNode.lchild = node
                return
            else:
                queue.append(curNode.lchild)
            if curNode.rchild == None:
                curNode.rchild = node
                return
            else: 
                queue.append(curNode.rchild)

    #广度优先遍历
    def travel(self):
        queue = []
        #判断根节点是否存在
        if self.root is None:
            return
        else:
            queue.append(self.root)
        while queue:
            curNode = queue.pop(0)
            print(curNode.elem,end='	')
            if curNode.lchild is not None:
                queue.append(curNode.lchild)
            if curNode.rchild is not None:
                queue.append(curNode.rchild)
        print()
    #先序遍历 根 左 右
    def preOrder(self,root):
        if root is None:
            return
        else:
            print(root.elem,end='	')
            self.preOrder(root.lchild)
            self.preOrder(root.rchild)

    #中序遍历 左 根 右
    def inOrder(self,root):
        if root is None:
            return
        else:
            self.inOrder(root.lchild)
            print(root.elem,end='	')
            self.inOrder(root.rchild)
    
    #后序遍历 左 右 根
    def postOrder(self,root):
        if root is None:
            return
        self.postOrder(root.lchild)
        self.postOrder(root.rchild)
        print(root.elem,end='	')

            
if __name__ == '__main__':
    tree = Tree()
    tree.add('A')
    tree.add('B')
    tree.add('C')
    tree.add('D')
    tree.add('E')
    print('广度优先遍历')
    tree.travel()
    print('先序遍历')
    tree.preOrder(tree.root)
    print()
    print('中序遍历')
    tree.inOrder(tree.root)
    print()
    print('后序遍历')
    tree.postOrder(tree.root)

广度优先遍历
A       B       C       D       E
先序遍历
A       B       D       E       C
中序遍历
D       B       E       A       C
后序遍历
D       E       B       C       A