本文实例讲述了Python实现二叉树的常见遍历操作。分享给大家供大家参考,具体如下:
二叉树的定义
class TreeNode: def __init__(self, x): self.val = x self.left = None self.right = None
二叉树的前序遍历
递归
def preorder(root,res=[]): if not root: return res.append(root.val) preorder(root.left,res) preorder(root.right,res) return res
迭代
def preorder(root): res=[] if not root: return [] stack=[root] while stack: node=stack.pop() res.append(node.val) if node.right: stack.append(node.right) if node.left: stack.append(node,left) return res
二叉树的中序遍历
递归
def inorder(root,res=[]): if not root: return inorder(root.left,res) res.append(root.val) inorder(root.right,res) return res
迭代
def inorder(root): stack=[] node=root res=[] while stack or node: while node: stack.append(node) node=node.left node=stack.pop() res.append(node.val) node=node.right return res
二叉树的后序遍历
递归
def laorder(root,res=[]): if not root: return laorder(root.left,res) laorder(root.right,res) res.append(root.val) return res
迭代
def laorder(root): stack=[root] res=[] while stack: node=stack.pop() if node.left: stack.append(node.left) if node.right: stack.append(node.right) res.append(node.val) return res[::-1]
二叉树的层次遍历
迭代
def levelorder(root): queue=[root] res=[] while queue: node=queue.pop(0) if node.left: queue.append(node.left) if node.right: queue.append(node.right) res.append(node.val) return res
最后多说一句,小编是一名python开发工程师,这里有我自己整理了一套最新的python系统学习教程,包括从基础的python脚本到web开发、爬虫、数据分析、数据可视化、机器学习等。想要这些资料的可以关注小编,并在后台私信小编:“01”即可领取。