标签:right res self 宽度 二叉树 编号 节点 leetcode662
题目:
给定一个二叉树,编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树(full binary tree)结构相同,但一些节点为空。
每一层的宽度被定义为两个端点(该层最左和最右的非空节点,两端点间的null节点也计入长度)之间的长度。
思路:
为树的每个节点进行编号,根节点从1开始编号
若当前节点的编号为i,则其左孩子节点的编号为2i,其右孩子节点的编号为2i+1
对二叉树进行层序遍历,更新res=max(res,每层最后一个节点的编号-每层第一个节点的编号+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:
#编号,当前节点i,左孩子2*i,右孩子2*i+1
#BFS
if not root:
return 0
q=[(root,1)]
res=1
while q:
n=len(q)
for i in range(n):
cur,idx=q.pop(0)
if i==0:
tmp=idx
if i==n-1:
res=max(res,idx-tmp+1)
if cur.left:
q.append((cur.left,2*idx))
if cur.right:
q.append((cur.right,2*idx+1))
return res
标签:right,res,self,宽度,二叉树,编号,节点,leetcode662 来源: https://blog.csdn.net/jqq125/article/details/123130912
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。