前往闪闪の小窝以获得更好的阅读和评论体验
一、前景提要
yesterday……hard……🤐
二、题目
难度:中等
给你一个二叉树的根节点 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, 104]范围内
- -105 <= Node.val <= 105
三、解答
1,非官方解法
BFS,一层一层遍历
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
int maxSum = Integer.MIN_VALUE;
int minFloor = 0;
// 一层一层遍历
public int maxLevelSum(TreeNode root) {
// 建一个队列用于存储每一层的所有节点
Deque<TreeNode> queue = new ArrayDeque<>();
queue.offer(root);
// 用于记录层号
int floor = 0;
while(!queue.isEmpty()){
floor++;
// 用于记录节点个数
int size = queue.size();
// 用于记录层和
int sum = 0;
// 当队列里不止一个节点时
while(size-- != 0){
TreeNode cur = queue.poll();
// 求和
sum += cur.val;
// 节点不为空时把它们加入队列等着下次一起求和
if(cur.left != null) queue.offer(cur.left);
if(cur.right != null) queue.offer(cur.right);
}
// 不用等于号,这样当出现多个相同的sum值的时候,minFloor的值就永远最小
if(sum > maxSum){
maxSum = sum;
minFloor = floor;
}
}
return minFloor;
}
}
2,官方解法
官方解法
BFS和DFS