Python将有序数组转换为二叉排序树的递归实现
发布时间：2026-10-13 |
6人浏览过
本文介绍如何使用Python将有序数组转换为二叉排序树。核心思路是递归选取数组中间元素作为根节点，分别构建左右子树。文章包含算法原理、图示说明及完整的Python代码实现，涵盖节点定义、递归构建及前中后序遍历验证，帮助开发者理解二叉树的结构特性与实现细节。
题目描述与二叉排序树定义
题目要求将包含元素[0,1,2,3,4,5,6,7,8,9,10]的有序数组存储到二叉树中，由于原数组已排序，转换后的二叉树应满足二叉排序树的性质。
二叉排序树（Binary Sort Tree）具有如下特性：对于任意节点，其左子树上所有节点的值均小于该节点的值，而右子树上所有节点的值均大于该节点的值。
二叉排序树也常被称为二叉查找树（Binary Search Tree），是数据结构中常见的一种树形结构。
递归转换思路与图示
实现该转换的具体思路如下：
采用递归策略，每次选取有序数组的中间元素作为当前子树的根节点，从而将数组划分为左半部分和右半部分。随后对左右两个子数组重复此操作，直至处理完所有元素。
转换过程的示意图如下所示：
图1展示了初始状态。
图2展示了中间步骤。
图3展示了最终生成的树结构。
Python代码实现与验证
以下是具体的Python代码实现：
def array_to_bitree(array):
#判断arr是否为空
if len(array)==0:
return BiTNode(array[0])
mid=len(array)//2 # 有序数组的中间元素的下标
#print(mid)
#start=0 # 数组第一个元素的下标
#end=-1 # 数组最后一个元素的下标
if len(array)>0:
#将中间元素作为二叉树的根
root=BiTNode(array[mid])
#如果左边的元素个数不为零，则递归调用函数，生成左子树
if len(array[:mid])>0:
root.left_child = arrayToBiTree(array[:mid])
#如果右边的元素个数不为零，则递归调用函数，生成左子树
if len(array[mid+1:])>0:
root.right_child = arrayToBiTree(array[mid+1:])
return root为了验证构造的二叉树是否正确，我们调用之前定义的三种遍历方法（前序、中序、后序）对生成的树进行遍历测试。
#将[0,1,2,3,4,5,6,7,8,9,10]存储到二叉树
if __name__ == '__main__':
#先构造一个有序数组、链表
arr=[]
for i in range(10):
arr.append(i)
print(arr)
#调用函数
BT=arrayToBiTree(arr)
#前序遍历二叉树
print("前序")
print_tree_pre_order(BT)
# 中序遍历二叉树
print("中序")
print_tree_mid_order(BT)
# 后序遍历二叉树
print("后序")
print_tree_after_order(BT)程序运行后的输出结果如下：
通过对比三种遍历结果，可以确认二叉树的结构符合预期，且与理论分析一致。具体代码实现如下：
#定义二叉树结点类型
class BiTNode:
"""docstring for BiTNode"""
def __init__(self,arg):
self.data = arg
self.left_child = None
self.right_child = None
#前序遍历
def print_tree_pre_order(root):
#先判断二叉树是否为空
#if root.left_child is None and root.right_child is None:
if root is None:
return root
#先根
print(root.data)
#再左
if root.left_child is not None:
print_tree_pre_order(root.left_child)
#再右
if root.right_child is not None:
print_tree_pre_order(root.right_child)
#中序遍历二叉树
def print_tree_mid_order(root):
#先判断二叉树是否为空,当左右节点都为空时
if root is None:
return
#中序遍历 左根右
#遍历左子树
if root.left_child is not None:
print_tree_mid_order(root.left_child)
#遍历根节点
print(root.data)
#遍历右子树
if root.right_child is not None:
print_tree_mid_order(root.right_child)
#后序遍历
def print_tree_after_order(root):
#先判断二叉树是否为空
if root is None:
return root
#再左
if root.left_child is not None:
print_tree_after_order(root.left_child)
#再右
if root.right_child is not None:
print_tree_after_order(root.right_child)
#先根
print(root.data)
def array_to_bitree(array):
#判断arr是否为空
if len(array)==0:
return BiTNode(array[0])
mid=len(array)//2 # 有序数组的中间元素的下标
#print(mid)
#start=0 # 数组第一个元素的下标
#end=-1 # 数组最后一个元素的下标
if len(array)>0:
#将中间元素作为二叉树的根
root=BiTNode(array[mid])
#如果左边的元素个数不为零，则递归调用函数，生成左子树
if len(array[:mid])>0:
root.left_child = array_to_bitree(array[:mid])
#如果右边的元素个数不为零，则递归调用函数，生成左子树
if len(array[mid+1:])>0:
root.right_child = array_to_bitree(array[mid+1:])
return root
#将[0,1,2,3,4,5,6,7,8,9,10]存储到二叉树
if __name__ == '__main__':
#先构造一个有序数组、链表
arr=[]
for i in range(9):
arr.append(i)
print(arr)
#调用函数
BT=array_to_bitree(arr)
#前序遍历二叉树
print("前序")
print_tree_pre_order(BT)
# 中序遍历二叉树
print("中序")
print_tree_mid_order(BT)
# 后序遍历二叉树
print("后序")
print_tree_after_order(BT)以上就是本文关于将有序数组转换为二叉树的全部内容，希望对相关学习有所帮助，欢迎继续支持本站获取更多技术资源。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
