leetcode题刷250天(96)——662. 二叉树最大宽度(BFS)

给你一棵二叉树的根节点 root ,返回树的 最大宽度 。
每一层的 宽度 被定义为该层最左和最右的非空节点(即,两个端点)之间的长度。

解题思路:
一棵完全二叉树可以按照一定的规则连续存储在数组中,其存储规则是从节点(i)开始,每个节点的左右孩子节点下标为(2i)以及(2i+1),按照此规则我们进行广度遍历,并以组合的形式存储节点的同时存储其位置信息,每遍历一层我们用最后一个节点位置减去第一个节点出现的位置,比较大小即可

# 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:
        que = deque()
        que.append( (root, 1) )
        maxWidth = 0

        while que:
            queLength = len(que)
            level = []
            
            for _ in range(queLength):
                node, position = que.popleft()
                level.append(position)
				# 其存储规则是从节点(i)开始,每个节点的左右孩子节点下标为(2*i)以及(2*i+1)
                if node is not None:
                    if node.left:
                        que.append( (node.left, 2*position) )
                    if node.right:
                        que.append( (node.right, 2*position+1) )
			# 用最后一个节点位置减去第一个节点出现的位置
            if (level[-1]-level[0]+1) > maxWidth:
                maxWidth = (level[-1]-level[0]+1)

        return maxWidth
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值