python遍历树结构_python数据结构之树（二叉树的遍历）

'''树的构造

1.递归实现先序遍历、中序遍历、后序遍历

2.堆栈实现先序遍历、中序遍历、后序遍历

3.队列实现层次遍历'''

#节点类

classNode(object):__slots__ = 'item','lchild','rchild'

def __init__(self,item=None,lchild=None,rchild=None):

self.item=item

self.lchild=lchild

self.rchild=rchild#树类

classTree(object):def __init__(self):

self.root=root

self.myQueue=myQueue#添加树节点

defadd(self,item):

node=Node(item)if self.root.item == None: #空则赋值root

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)#如果该节点在右子树，丢弃该节点

#递归实现树的先序遍历

deffront_digui(self,root):if root ==None:returnNoneprint(root.item)

self.front_digui(root.lchild)

self.front_digui(root.rchild)#递归实现树的中序遍历

defmiddle_digui(self,root):if root ==None:returnNone

self.middle_digui(root.lchild)print(root.item)

self.middle_digui(root.rchild)#递归实现树的后序遍历

deflater_digui(self,root):if root ==None:returnNone

self.later_digui(root.lchild)

self.later_digui(root.rchild)print(root.item)#利用堆栈实现树的先序遍历

deffront_stack(self, root):if root ==None:returnmyStack=[]

node=rootwhile node ormyStack:while node: #从根节点开始，一直找它的左子树

print(node.item)

myStack.append(node)

node=node.lchild

node= myStack.pop() #while结束表示当前节点node为空，即前一个节点没有左子树了

node = node.rchild #开始查看它的右子树

#利用堆栈实现树的中序遍历

defmiddle_stack(self, root):if root ==None:returnmyStack=[]

node=rootwhile node ormyStack:while node: #从根节点开始，一直找它的左子树

myStack.append(node)

node=node.lchild

node= myStack.pop() #while结束表示当前节点node为空，即前一个节点没有左子树了

printnode.item,

node= node.rchild #开始查看它的右子树

#利用堆栈实现树的后序遍历

deflater_stack(self, root):if root ==None:returnmyStack1=[]

myStack2=[]

node=root

myStack1.append(node)while myStack1: #这个while循环的功能是找出后序遍历的逆序，存在myStack2里面

node =myStack1.pop()ifnode.lchild:

myStack1.append(node.lchild)ifnode.rchild:

myStack1.append(node.rchild)

myStack2.append(node)while myStack2: #将myStack2中的元素出栈，即为后序遍历次序

print(myStack2.pop().item)#利用队列实现树的层次遍历

deflevel_queue(self, root):if root ==None:returnmyQueue=[]

node=root

myQueue.append(node)whilemyQueue:

node=myQueue.pop(0)printnode.item,if node.lchild !=None:

myQueue.append(node.lchild)if node.rchild !=None:

myQueue.append(node.rchild)#测试

if __name__ == '__main__':

items= range(10) #生成十个数据作为树节点

tree = Tree() #新建一个树对象

for item initems:

tree.add(item)#逐个添加树的节点

print('队列实现层次遍历:')

tree.level_queue(tree.root)print('\n\n递归实现先序遍历:')

tree.front_digui(tree.root)print('\n递归实现中序遍历:')

tree.middle_digui(tree.root)print('\n递归实现后序遍历:')

tree.later_digui(tree.root)print('\n\n堆栈实现先序遍历:')

tree.front_stack(tree.root)print('\n堆栈实现中序遍历:')

tree.middle_stack(tree.root)print('\n堆栈实现后序遍历:')

tree.later_stack(tree.root)