其他分享
首页 > 其他分享> > [LC1161]最大层内元素和

[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