给你一棵二叉树的根节点 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