[小白系列]Python实现逻辑树、遍历树（深度优先以及广度优先），以非计算机专业的角度学习把数组放入二叉树

1
2
3
4
5
6
class BiNode:
    def __init__(self):
        self.data = None   #结点数据
        self.lchild = None #左子树结点
        self.rchild = None #右子树结点

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#将有序数组转换为二叉树
def array_to_tree(arr, start, end):
    """将有序数组转换成二叉树
        arr：有序数组
        start：数组起点索引
        end：数组结尾索引
        """
    root = None  #初始化根结点数据
    if end >= start:
        root = BiNode() #创建树实例
        mid = (start + end + 1)//2  #取中点数据索引
        root.data = arr[mid] #树的根结点诶数组中间元素
        root.lchild = array_to_tree(arr, start, mid - 1)  #使用递归方式用左半部分数组构建root的左子树
        root.rchild = array_to_tree(arr, mid+1, end) #使用递归方式用右半部分数组构建root的右子树
    else:
        root = None
    return root

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#使用中序遍历的方式打印二叉树
def print_tree_midorder(root):
    """使用中序遍历的方式打印二叉树内容
        root：二叉树根结点
    """
    if root == None: #判断是否为空
        return
        
    #使用递归方式，遍历root节点的左子树
    if root.lchild != None:
        print_tree_midorder(root.lchild)
            
    #遍历root节点
    print(root.data, end = ' ')
        
    #使用递归方式，遍历root节点右子树
    if root.rchild != None:
        print_tree_midorder(root.rchild)

1
2
3
arr = list(range(1,11))
arr

1
2
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

1
2
root = array_to_tree(arr, 0, len(arr)-1)

1
2
3
print('树的中序遍历结果为：', end = '')
print_tree_midorder(root)

1
2
树的中序遍历结果为：1 2 3 4 5 6 7 8 9 10 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#使用层序遍历的方式打印二叉树
from collections import deque  #载入队列结构,deque是双向队列
def print_tree_layer(root):
    """使用层序遍历的方式打印二叉树内容
        root：二叉树根结点
    """
    if root == None:
        return
        
    queue = deque() #创建队列实例
    #树根结点进队列
    queue.append(root)
        
    while len(queue)>0:
        p = queue.popleft() #出队
        #访问当前节点
        print(p.data, end = ' ')
            
        #如果该节点左孩子不为空，则入队列
        if p.lchild != None:
            queue.append(p.lchild)
                
        #如果该节点右孩子不为空，则入队列
        if p.rchild != None:
            queue.append(p.rchild)

1
2
3
arr = list(range(1,11))
arr

1
2
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

1
2
root = array_to_tree(arr, 0, len(arr)-1)

1
2
3
print('树的层序遍历结果为：', end = '')
print_tree_layer(root)

1
2
树的层序遍历结果为：6 3 9 2 5 8 10 1 4 7