[LC1161]最大层内元素和
作者:互联网
题目概述
给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层,而根节点的子节点位于第 2 层,依此类推。
请返回层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/maximum-level-sum-of-a-binary-tree
树中的节点数在 [1, 10^4]范围内,
-10^5 <= Node.val <= 10^5
题目分析
一道经典的树的遍历问题,解决问题是可以采用深度优先遍历,也可以采用层序遍历。深度优先的函数接口设计中,需要传入层级level,并使用数组来缓存每一层的和。层序遍历中,借助队列结构,即可实现。
代码
层序遍历-golang
func maxLevelSum(root *TreeNode) int {
resLevel, maxSum := 1, math.MinInt32
queue := []*TreeNode{root}
for level := 1; queue != nil; level++ {
tmpQueue := queue
queue = nil
levelSum := 0
for _, node := range tmpQueue {
levelSum += node.Val
if node.Left != nil {
queue = append(queue, node.Left)
}
if node.Right != nil {
queue = append(queue, node.Right)
}
}
if levelSum > maxSum{
maxSum = levelSum
resLevel = level
}
}
return resLevel
}
标签:node,层内,遍历,nil,level,元素,queue,LC1161,levelSum 来源: https://www.cnblogs.com/xy1997/p/16536556.html