python实现二叉树的建立以及遍历（递归前序、中序、后序遍历，队栈前序、中序、后序、层次遍历）

#-*- coding:utf-8 -*-
class Node:
    def __init__(self,data):
        self.data=data
        self.lchild=None
        self.rchild=None

class Tree:
    def __init__(self):
        self.queue=[]#利用队列存储树的节点
        self.flag=0#存储树根后flag置为1
        self.root=None

    #建树
    def createTree(self,list):
        while True:
            #list中没有数据，表示建树完成
            if len(list)==0:
                return
            #flag为0，表示树根不存在
            if self.flag==0:
                self.root=Node(list[0])
                #讲树根存入队列
                self.queue.append(self.root)
                #树根已创建，flag置为1
                self.flag=1
                #剔除list中第一个已经使用数
                list.pop(0)
            else:
                '''
                treeNode:队列中的第一个节点(该节点左右孩子不完全存在)
                添加treeNode的左右孩子，当添加treeNode的右孩子之后，
                将队列中的第一个节点出队。
                '''
                treeNode=self.queue[0]
                if treeNode.lchild==None:
                    treeNode.lchild=Node(list[0])
                    self.queue.append(treeNode.lchild)
                    list.pop(0)
                else:
                    treeNode.rchild = Node(list[0])
                    self.queue.append(treeNode.rchild)
                    list.pop(0)
                    self.queue.pop(0)


    # 递归实现先序遍历
    def front_digui(self,root):
        if root==None:
            return
        else:
            print root.data,
            self.front_digui(root.lchild)
            self.front_digui(root.rchild)
    # 递归实现中序遍历
    def middle_digui(self,root):
        if root==None:
            return
        else:
            self.middle_digui(root.lchild)
            print root.data,
            self.middle_digui(root.rchild)
    # 递归实现后序遍历
    def behind_digui(self,root):
        if root==None:
            return
        else:
            self.behind_digui(root.lchild)
            self.behind_digui(root.rchild)
            print root.data,

    # 队栈实现先序遍历
    def front_queueAndStack(self,root):
        if root==None:
            return
        #定义一个栈,存储节点
        stack=[]
        node=root
        while stack or node:
            #从树根开始一直输出左孩子
            while node:
                print node.data,
                #将输出的节点加入栈中
                stack.append(node)
                node=node.lchild
            #该节点不存在左节点时,该节点出栈,搜索该节点右节点，
            node=stack.pop()
            node=node.rchild
    # 队栈实现中序遍历
    def middle_queueAndStack(self,root):
        if root==None:
            return
        # 定义一个栈,存储节点
        stack = []
        node = root
        while stack or node:
            #一直查找树的左节点,一直进栈
            while node:
                stack.append(node)
                node=node.lchild
            node=stack.pop()#该节点不存在左节点，该节点出栈，查找右节点
            print node.data,
            node=node.rchild
    # 队栈实现后序遍历
    def behind_queueAndStack(self,root):
        if root==None:
            return
        # 定义一个栈,存储节点
        stack_1 = []
        stack_2 = []
        node = root
        stack_1.append(node)
        while stack_1:
            #该节点出栈1.左右节点进栈1(对于左右节点,右节点先出栈1,也先进栈1)
            node=stack_1.pop()
            if node.lchild:
                stack_1.append(node.lchild)
            if node.rchild:
                stack_1.append(node.rchild)
            #该节点进栈2
            stack_2.append(node)
        while stack_2:
            print stack_2.pop().data,
    # 队栈实现层次遍历
    def level_queueAndStack(self,root):
        if root==None:
            return
        stack_1=[]
        stack_2=[]
        stack_1.append(root)
        stack_2.append(root)
        while stack_1:
            node=stack_1.pop(0)
            if node.lchild:
                stack_1.append(node.lchild)
                stack_2.append(node.lchild)
            if node.rchild:
                stack_1.append(node.rchild)
                stack_2.append(node.rchild)
        while stack_2:
            print stack_2.pop(0).data,


if __name__ == '__main__':
    list=[0,1,2,3,4,5,6,7,8,9,]
    tree=Tree()
    tree.createTree(list)
    tree.front_digui(tree.root)
    print '
'
    tree.middle_digui(tree.root)
    print '
'
    tree.behind_digui(tree.root)
    print '
'
    tree.front_queueAndStack(tree.root)
    print '
'
    tree.middle_queueAndStack(tree.root)
    print '
'
    tree.behind_queueAndStack(tree.root)
    print '
'
    tree.level_queueAndStack(tree.root)