Programmer-3.2 如何把一个有序整数数组放到二叉树中

2024年6月10日 98次阅读

题目描述：

如何把一个有序整数数组放到二叉树中

分析与解答：

取数组的中间元素作为根节点，将数组分成左右两部分，对数组的两部分用递归的方法分别构建左右子树。

# 实现队列
class BiTNode:
    def __init__(self):
        self.data = None
        self.lchild = None
        self.rchild = None

# 方法功能：把有序数组转换成二叉树
def arraytotree(arr, start, end):
    root = None
    if end>=start:
        root = BiTNode()
        mid = (start+end+1)//2
        # 树的根节点为数组中间的元素
        root.data = arr[mid]
        # 递归的用左半部分数组构造root左子树
        root.lchild = arraytotree(arr, start, mid-1)
        # 递归的用右半部分数组构造root右子树
        root.rchild = arraytotree(arr, mid+1, end)
    else:
        root = None
    return root

# 用中序遍历的方式打印出二叉树结点的内容
def printTreeMidOrder(root):
    if root == None:
        return
    # 遍历root结点的左子树
    if root.lchild != None:
        printTreeMidOrder(root.lchild)
    # 遍历root结点
    print(root.data)
    # 遍历root结点的右子树
    if root.rchild != None:
        printTreeMidOrder(root.rchild)

if __name__=="__main__":
    arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    print("数组：")
    i = 0
    while i<len(arr):
        print(arr[i])
        i += 1
    # 把有序数组转换成二叉树
    root = arraytotree(arr, 0, len(arr)-1)
    print("转换成树的中序遍历为：")
    printTreeMidOrder(root)

运行结果：

数组：
1
2
3
4
5
6
7
8
9
10
转换成树的中序遍历为：
1
2
3
4
5
6
7
8
9
10

点赞