515_在每个树行中找最大值

JaysenLeo

于 2024-01-16 21:30:15 发布

阅读量450

点赞数 8

分类专栏：激进的算法文章标签： python 算法广度优先

本文为博主原创文章，转载文章须附上文章来源与作者。

本文链接：https://blog.youkuaiyun.com/ChangerJJLee/article/details/135635913

版权

激进的算法专栏收录该内容

12 篇文章

订阅专栏

文章介绍了如何使用层遍历的方法，通过维护一个队列来获取给定二叉树中每一层的最大值，代码展示了如何在Python中实现Solution类的largestValues方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述
给定一棵二叉树的根节点 root ，请找出该二叉树中每一层的最大值
思路
1.层遍历时记录最大值
解答

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def largestValues(self, root: Optional[TreeNode]) -> List[int]:
        if root is None:
            return []
        ret = []
        q=[root, ]
        while q:
            curr_level = q
            nxt_level = []
            max_val = -inf
            for node in curr_level:
                if node:
                    max_val = max(max_val, node.val)
                    node.left and nxt_level.append(node.left) or None
                    node.right and nxt_level.append(node.right) or None
            q = nxt_level
            ret.append(max_val)
        return ret