LeetCode 662 二叉树最大宽度

最新推荐文章于 2024-08-16 11:04:03 发布

原创最新推荐文章于 2024-08-16 11:04:03 发布 · 180 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #二叉树 #Golang

leetcode 专栏收录该内容

64 篇文章

订阅专栏

本文介绍了一种使用层次遍历的方法计算二叉树最大宽度，通过数组存储节点并利用节点在当前层的序号进行扩展，关键在于计算时注意加上额外的1。算法详细展示了如何利用item结构体跟踪节点序号并更新宽度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这里采用同层次遍历相同的策略，同样是用数组存放结点，区别在于增加了结点所在当前层的序号，这个根据父节点的序号计算（左节点是父节点序号的2倍，右节点是父节点序号的2倍+1）。

需要注意的是，计算宽度的时候别忘了➕1！！！

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
// 借助一个记录序列号的结构体来计算二叉树的最大宽度
 type item struct{
     idx int
     *TreeNode
 }

func widthOfBinaryTree(root *TreeNode) int {
    if root == nil {
        return 0
    }
    // 初始宽度为1
    ans, q := 1, []item{{0,root}}
    for len(q)>0 {
        // 计算当前层的最大宽度，注意还要加一
        if l := q[len(q)-1].idx - q[0].idx + 1; l > ans {
            ans = l
        }
        // 存放下一层的结构体（根据当前层结点的索引去计算左右子节点上的索引编号）
        temp := []item{}
        for _, p := range q{
            if p.Left != nil {
                temp = append(temp, item{p.idx*2, p.Left})
            } 
            if p.Right != nil {
                temp = append(temp, item{p.idx*2+1, p.Right})
            }
        }
        q = temp
    }
    return ans
}