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

转载：https://www.cnblogs.com/ybf-yyj/p/8717601.html

见二叉树先想递归。

            
              -*- 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 '\n'
    tree.middle_digui(tree.root)
    print '\n'
    tree.behind_digui(tree.root)
    print '\n'
    tree.front_queueAndStack(tree.root)
    print '\n'
    tree.middle_queueAndStack(tree.root)
    print '\n'
    tree.behind_queueAndStack(tree.root)
    print '\n'
    tree.level_queueAndStack(tree.root)