[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
}