题解 | #重量级的一层#
重量级的一层
https://www.nowcoder.com/practice/193372871b09426ab9ea805f0fd44d5c
- 题目考察的知识点
二叉树的层次遍历
- 题目解答方法的文字分析
首先定义 maxheight来保存牛的最大总重量,定义ans为具有牛的最大总重量的那一层的层数。因为牛的总重量和层数都不可能为负数,为此,初始化maxheight和ans为-1。然后层次遍历二叉树,用sum来保存当前层的牛的总重量,用ceng来保存当前层数。当sum>=maxheight时,修改maxheight和ans的值。当二叉树遍历完后,ans就是答案。
- 本题解析所用的编程语言
java
- 完整且正确的编程代码
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型
*/
public int maxLevelSum (TreeNode root) {
int maxheight = -1;
int ans = -1;
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
int ceng = 0;
while(!queue.isEmpty()){
int sum = 0;
int count = queue.size();
ceng++;
while(count-- >0){
TreeNode node = queue.poll();
sum+=node.val;
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
if(maxheight<=sum){
ans=ceng;
maxheight = sum;
}
}
return ans;
}
}