二叉树的遍历 python实现

最新推荐文章于 2024-08-24 08:16:26 发布

原创最新推荐文章于 2024-08-24 08:16:26 发布 · 269 阅读

0 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

26 篇文章

订阅专栏

本文深入探讨了二叉树的四种主要遍历方法：先序遍历、中序遍历、后序遍历和层次遍历。通过递归和非递归方式详细解释了每种遍历的实现原理，并介绍了如何计算二叉树的节点数量和最大深度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

先序遍历——根节点→左子树→右子树

递归形式：

def preorder(self,root):
	if root == None:
		return 
	print(root.val)
	preorder(root.left)
	preorder(root.right)

非递归形式：（利用栈的性质）

def preorder(self,root):
	stack = [root]
	while len(stack)>0:
		print(root.val)
		if root.right is not None:
			stack.append(root.right)
		if root.left is not None:
			stack.append(root.left)
		root =stack.pop()

中序遍历——左子树→根节点→右子树

递归形式：

def inorder(self, root):
	if root == None:
		return
	inorder(root.left)
	print(root.val)
	inorder(root.right)

非递归：

#先解决左节点，后解决右节点，根节点顺带。
def inorder(self,root):
	stack=[]
	pos=root
	while pos is not None or len(stack)>0:
		if pos is not None:
			stack.append(pos)
			pos = pos.left
		else:
			pos = stack.pop()
			print(pos.val)
			pos=pos.right

后序遍历——左子树→右子树→根节点

递归：

def  postorder(self,root):
	if root ==None:
		return 
	postorder(root.left)
	postorder(root.right)
	print(root.val)

非递归:

# 使用两个栈结构
# 第一个栈进栈顺序：左节点->右节点->跟节点
# 第一个栈弹入第二个栈顺序： 跟节点->右节点->左节点
# 第二个栈的弹出顺序：左节点->右节点->根节点
def postorder(self,root):
	stack=[root]
	stack2=[]
	while len(stack)>0:
		root = stack.pop()
		stack2.append(root)
		if root.left is not None:
			stack.append(root.left)
		if root.right is not None:
			stack.append(root.right)
	while len(stack2)>0:
		print(stack2.pop().val)

层次遍历（从上到下，从左到右遍历）

def breadth_travel(self, root):
	if root ==None:
		return
	queue =[]
	queue.append(root)
	while queue:
		node = queue.pop(0)
		print  (node.val)
		if node.left:
			queue.append(node.left)
		if node.right:
			queue.append(node.right)

二叉树节点个数

 def  treeNode(node):
 	if node is None:
 		return 0
 	nums =treeNode(node.left)
 	nums += treeNode(node.right)
 	return nums+1

二叉树的最大深度

def treeDepth(node):
	if node is None:
		return 0
	ldepth = treedepth(node.left)
	rdepth = treedepth(node.right)
	return (max(ldepth,rdepth)+1)