左神算法系列(Python代码)：二叉树套路题

smashed_potato

已于 2022-04-20 05:27:07 修改

阅读量1k

点赞数 1

分类专栏：算法刷题文章标签： python 数据结构算法

于 2022-04-20 05:26:16 首次发布

本文链接：https://blog.csdn.net/weixin_39043344/article/details/124287977

版权

本文介绍如何解决二叉树的最大宽度问题，通过层序遍历的方法，详细讲解了两种策略：1）利用HashMap记录节点所在层级；2）使用额外变量curEnd和nextEnd判断层末。这两种方法都涉及到信息传递，帮助判断是否达到当前层的结尾。在实际应用中需注意变量初始化和max_更新的时机。

摘要由CSDN通过智能技术生成

class Node:
    def __init__(self, value, left=None, right=None):
        self.value = value
        self.left = left
        self.right = right
        self.parent = None


def get_max_width_with_map(node):
    print('The max width of the tree is: ', end=" ")
    if node is not None:
        queue = []
        level_map = {} # 记录每一个节点的所处层
        queue.append(node) # 放入头节点
        level_map[node] = 1 # 设置头节点所在层
        cur_level = 1 # accumulator -> 记录遍历的层数
        cur_level_nodes = 0 # 储存当前层所遍历的节点数
        max_ = 0
        while len(queue) > 0:
            cur_node = queue.pop(0)
            cur_node_level = level_map[cur_node]
            # 分别遍历左右两个节点，并保存其所在层数：当前节点层+1
            if cur_node.left is not None: 
                queue.append(cur_node.left)
                level_map[cur_node.left] = cur_node_level + 1
            if cur_node.righ