题解 | #重量级的一层#

重量级的一层

https://www.nowcoder.com/practice/193372871b09426ab9ea805f0fd44d5c

  1. 题目考察的知识点

二叉树的层次遍历

  1. 题目解答方法的文字分析

首先定义 maxheight来保存牛的最大总重量,定义ans为具有牛的最大总重量的那一层的层数。因为牛的总重量和层数都不可能为负数,为此,初始化maxheight和ans为-1。然后层次遍历二叉树,用sum来保存当前层的牛的总重量,用ceng来保存当前层数。当sum>=maxheight时,修改maxheight和ans的值。当二叉树遍历完后,ans就是答案。

  1. 本题解析所用的编程语言

java

  1. 完整且正确的编程代码
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;

    }
}
全部评论

相关推荐

我是小红是我:学校换成中南
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务