二叉树初始化

SZ170110231

已于 2022-11-10 19:53:21 修改

阅读量696

点赞数 3

文章标签： python

于 2022-11-10 19:47:23 首次发布

本文链接：https://blog.csdn.net/SZ170110231/article/details/127795111

版权

node_dict:

在这里插入图片描述

二叉树效果图：

在这里插入图片描述

node_list = [
    {'data': 'A', 'left': 'B', 'right': 'C', 'is_root': True},
    {'data': 'B', 'left': 'D', 'right': 'E', 'is_root': False},
    {'data': 'D', 'left': None, 'right': None, 'is_root': False},
    {'data': 'E', 'left': 'H', 'right': None, 'is_root': False},
    {'data': 'H', 'left': None, 'right': None, 'is_root': False},
    {'data': 'C', 'left': 'F', 'right': 'G', 'is_root': False},
    {'data': 'F', 'left': None, 'right': None, 'is_root': False},
    {'data': 'G', 'left': 'I', 'right': 'J', 'is_root': False},
    {'data': 'I', 'left': None, 'right': None, 'is_root': False},
    {'data': 'J', 'left': None, 'right': None, 'is_root': False},
]

class Node:
    def __init__(self,data,left=None,right=None):
        self.data = data
        self.left = left #Node指针而非数值
        self.right = right #Node指针而非数值

class Tree:
    def __init__(self,root=None):
        self.root = root #指向根节点
    def init_data(self,node_list):
        node_dict = {}
        #node_dict
        #key: item['data']
        #value: Node(item['data'],item['left'],item['right']) 
        for item in node_list:
            #第一次：item = {'data': 'A', 'left': 'B', 'right': 'C', 'is_root': True}
            #封装成节点,新增到node_dict
            node = Node(item['data'],item['left'],item['right'])
            node_dict[item['data']] = node
            #            ^             ^
            #          key(data)    value(Node)

            if item['is_root']:
                self.root = node_dict[item['data']]
            #node_dict的消失
            #引用计数: 任何内存, 一旦没有指向它的引用, 就会立即被回收
            #函数返回，函数创建变量node_dict被回收
            #但node_dict指向的内容被 self.root指向，并不会被回收

        '''
        注释''' '''也要严格按照缩进要求
        # 字典遍历 修改Value：(可能会很慢)
        # 如果新增键值对，导致扩容，键值对会重新排序
        for key in node_dict.keys():
            print(key,node_dict[key])
            node = node_dict[key]
            #node.left != None
            #node.left存储下一节点取值
            if node.left:
                #将取值转换为指向取值的节点
                node.left = node_dict[node.left]
            if node.right:
                node.right = node_dict[node.right]
        '''

        #列表转换后得到字典
        #根据列表中元素值对 字典key 遍历
        for item in node_list:
        #for item in node_list:
            # item['data'] <--> key
            node = node_dict[item['data']]
            if node.left:
                #将取值转换为取值节点的指向
                node.left = node_dict[node.left]
            if node.right:
                node.right = node_dict[node.right]

if __name__ == "__main__":
    
    tree = Tree()
    tree.init_data(node_list)