题目
原题链接:1161. 最大层内元素和
给你一个二叉树的根节点 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
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
|
class Solution { public int maxLevelSum(TreeNode root) { Queue<TreeNode> queue = new LinkedList<>(); queue.add(root); int ans = 1; int max = Integer.MIN_VALUE; int level = 1; while(!queue.isEmpty()){ int n = queue.size(); int sum = 0; for(int i = 0 ; i < n ; i++){ TreeNode node = queue.poll(); sum += node.val; if(node.left != null) { queue.add(node.left); } if(node.right != null) { queue.add(node.right); } } if(sum > max){ max = sum; ans = level; } level++; } return ans; } }
|
层序遍历的同时多一个计数和比较判断就行了。