Leetcode 1161. 最大层内元素和(Golang解法)

Leetcode 1161. 最大层内元素和(Golang解法)

题目

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。

请你找出层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个。

示例 1：

输入：root = [1,7,0,7,-8,null,null]
输出：2
解释：
第 1 层各元素之和为 1，
第 2 层各元素之和为 7 + 0 = 7，
第 3 层各元素之和为 7 + -8 = -1，
所以我们返回第 2 层的层号，它的层内元素之和最大。

示例 2：

输入：root = [989,null,10250,98693,-89388,null,null,null,-32127]
输出：2

思路

• 创建一个队列
• 对每层的数据累加求和， 若当前层的最大值超过记录的最大值则更新层数

注：
golang创建队列的方法：
利用切片创建队列
var queue []*TreeNode = []*TreeNode{root} // 定义一个队列
队尾插入元素
queue = append(queue, front.Left)
queue = append(queue, front.Right)
队头删除元素
queue = queue[1:len(queue)]

代码 —— golang

func maxLevelSum(root *TreeNode) int {
	if root == nil {
		return 0
	}

	var queue []*TreeNode = []*TreeNode{root} // 定义一个队列
	res, max, floor := -1, math.MinInt32, 0

	for len(queue) > 0 {
		size, sum := len(queue), 0
		floor++

		for i := 0;i < size;i++ {
			front := queue[0]
			queue = queue[1:len(queue)]

			sum += front.Val

			if front.Left != nil {
				queue = append(queue, front.Left)
			}

			if front.Right != nil {
				queue = append(queue, front.Right)
			}
		}

		if sum > max {
			res = floor
            max = sum
		}
	}

	return res
}