#!/usr/bin/python3#coding:utf-8
'''Created on 2018/09/14 10:48
@author: 刘蒙华'''
classNode():def __init__(self, data=-1, lchild=None, rchild=None):
self.data=data
self.lchild=lchild
self.rchild=rchildclassTree():def __init__(self):
self.root=Node()
self.myQueue=[]defadd(self, data):"""为树添加节点"""node=Node(data)if self.root.data == -1: #如果树是空的,则对根节点赋值
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:return
print(root.data)
self.front_digui(root.lchild)
self.front_digui(root.rchild)defmiddle_digui(self, root):"""利用递归实现树的中序遍历:
若二叉树为空,则空操作返回。否则从根结点开始(注意并不是先访问根结点),中序遍历根结点的左子树,然后是访问根结点,最后中序遍历右子树"""
if root ==None:returnself.middle_digui(root.lchild)print(root.data)
self.middle_digui(root.rchild)deflater_digui(self, root):"""利用递归实现树的后序遍历:
若二叉树为空,则空操作返回。否则从左到右先叶子后结点的方式遍历访问左右子树,最后访问根结点。"""
if root ==None:returnself.later_digui(root.lchild)
self.later_digui(root.rchild)print(root.data)deffront_stack(self, root):if root ==None:returnmyStack=[]
node=rootwhile node ormyStack:while node: #从根节点开始,一直找它的左子树
print(node.data)
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为空,即前一个节点没有左子树了
print(node.data)
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().data)deflevel_queue(self,root):'''利用队列实现数的层次遍历'''
if root ==None:returnmyQueue=[]
node=root
myQueue.append(node)whilemyQueue:
node=myQueue.pop(0)print(node.data)if node.lchild !=None:
myQueue.append(node.lchild)if node.rchild !=None:
myQueue.append(node.rchild)if __name__ == '__main__':
elems= range(10) #生成十个数据作为树节点
tree = Tree() #新建一个树对象
for elem inelems:
tree.add(elem)#逐个添加树的节点
print(tree.__dict__)
tree.middle_digui(tree.root)