2022年 11月 9日

Python实现二叉树的常见遍历操作总结「7种方法」

本文实例讲述了Python实现二叉树的常见遍历操作。分享给大家供大家参考,具体如下:

二叉树的定义

Python实现二叉树的常见遍历操作总结「7种方法」

class TreeNode: 
def __init__(self, x): 
self.val = x self.left = None 
self.right = None

二叉树的前序遍历

递归

Python实现二叉树的常见遍历操作总结「7种方法」

def preorder(root,res=[]): 
if not root: 
return res.append(root.val) 
preorder(root.left,res) 
preorder(root.right,res) 
return res

迭代

Python实现二叉树的常见遍历操作总结「7种方法」

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

二叉树的中序遍历

递归

Python实现二叉树的常见遍历操作总结「7种方法」

def inorder(root,res=[]): 
if not root: 
return inorder(root.left,res) 
res.append(root.val) 
inorder(root.right,res) 
return res

迭代

image.png

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

二叉树的后序遍历

递归

image.png

def laorder(root,res=[]): 
if not root: return laorder(root.left,res) 
laorder(root.right,res) 
res.append(root.val) 
return res

迭代

image.png

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]

二叉树的层次遍历

迭代

image.png

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”即可领取。