题解 | #重量级的一层#

重量级的一层

https://www.nowcoder.com/practice/193372871b09426ab9ea805f0fd44d5c?tpId=354&tqId=10595898&ru=/exam/oj/ta&qru=/ta/interview-202-top/question-ranking&sourceUrl=%2Fexam%2Foj%2Fta%3FtpId%3D354

知识点:

树的层序遍历

解题思路:

用一个queue存储每一层,遍历当前层的queue,一边算出sum一边将下一层节点加入queue。每遍历完一层就判断是否更新res

语言:

Golang

package main
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return int整型
*/
func maxLevelSum( root *TreeNode ) int {
    // write code here
    depth:=1
    queue:=[]*TreeNode{root}
    maxSum:=0
    res:=1
    for len(queue)>0{
        size:=len(queue)
        sum:=0
        for i:=0;i<size;i++{
            cur:=queue[0]
            sum+=cur.Val
            queue = queue[1:]
            if cur.Left != nil{
                queue = append(queue, cur.Left)
            }
            if cur.Right !=nil{
                queue = append(queue, cur.Right)
            }
        }
        if sum > maxSum{
            maxSum =sum
            res = depth
        }
        depth++
    }
    return res
}

全部评论

相关推荐

05-09 12:23
已编辑
华南理工大学 Java
野猪不是猪🐗:给他装的,双九+有实习的能看的上这种厂我直接吃⑨✌们拿它练练面试愣是给他整出幻觉了
点赞 评论 收藏
分享
犹豫的小狐狸刷了100道题:你是我在牛课上见到的最漂亮的女孩了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务