Python 实现二叉树前序，中序，后序，层次遍历

class Node(object):

“”“节点类”“”

def init(self, element=-1, l_child=None, r_child=None):

self.element = element

self.l_child = l_child

self.r_child = r_child

class Tree(object):

“”“树类”“”

def init(self):

self.root = Node()

self.queue = []

def add_node(self, element):

“”“为树添加节点”“”

node = Node(element)

如果树是空的，则对根节点赋值

if self.root.element == -1:

self.root = node

self.queue.append(self.root)

else:

tree_node = self.queue[0]

此结点没有左子树，则创建左子树节点

if tree_node.l_child is None:

tree_node.l_child = node

self.queue.append(tree_node.l_child)

else:

tree_node.r_child = node

self.queue.append(tree_node.r_child)

如果该结点存在右子树，将此节点丢弃

self.queue.pop(0)

def front_recursion(self, root):

“”“利用递归实现树的前序遍历”“”

if root is None:

return

print root.element,

self.front_recursion(root.l_child)

self.front_recursion(root.r_child)

def middle_recursion(self, root):

“”“利用递归实现树的中序遍历”“”

if root is None:

return

self.middle_recursion(root.l_child)

print root.element,

self.middle_recursion(root.r_child)

def back_recursion(self, root):

“”“利用递归实现树的后序遍历”“”

if root is None:

return

self.back_recursion(root.l_child)

self.back_recursion(root.r_child)

print root.element,

@staticmethod

def front_stack(root):

“”“利用堆栈实现树的前序遍历”“”

if root is None:

return

stack = []

node = root

while node or stack:

从根节点开始，一直找它的左子树

while node:

print node.element,

stack.append(node)

node = node.l_child

while结束表示当前节点node为空，即前一个节点没有左子树了

node = stack.pop()

开始查看它的右子树

node = node.r_child

@staticmethod

def middle_stack(root):

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

if root is None:

return

stack = []

node = root

while node or stack:

从根节点开始，一直找它的左子树

while node:

stack.append(node)

node = node.l_child

while结束表示当前节点node为空，即前一个节点没有左子树了

node = stack.pop()

print node.element,

开始查看它的右子树

node = node.r_child

@staticmethod

def back_stack(root):

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

if root is None:

return

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里无偿获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！