【重点】【二叉树】662.二叉树最大宽度

原创

已于 2025-06-28 13:09:03 修改 · 663 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#BFS #二叉树的最大宽度

于 2024-01-19 14:27:59 首次发布

本文介绍了两种方法解决二叉树宽度问题：一种是修改原TreeNode结构，另一种是构建新TreeNode。两种方法都使用BFS遍历，通过广度优先搜索计算二叉树的最大宽度。

题目

法1：BFS，最佳方法

在这里插入图片描述

版本1：修改原TreeNode

Python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def widthOfBinaryTree(self, root: Optional[TreeNode]) -> int:
        res = 0
        queue = collections.deque()
        root.val = 1
        queue.append(root)
        while queue:
            tmpSize = len(queue)
            start, end = -1, -1
            for i in range(tmpSize):
                curNode = queue.popleft()
                val = curNode.val
                if i == 0:
                    start = curNode.val
                if i == tmpSize - 1:
                    end = curNode.val
                if curNode.left:
                    curNode.left