515. Find Largest Value in Each Tree Row [Medium] BFS

最新推荐文章于 2021-06-02 12:01:06 发布

原创最新推荐文章于 2021-06-02 12:01:06 发布 · 127 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于解决寻找二叉树每一层最大值的问题。通过使用队列进行层次遍历，该算法能够有效地找到二叉树中每一层的最大值。

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

515. Find Largest Value in Each Tree Row

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def largestValues(self, root):
        """
        :type root: TreeNode
        :rtype: List[int]
        """
        if not root:
            return []
        qu = [root]
        res = []
        while qu:
            mx = float("-inf")
            for i in range(len(qu)):
                node = qu.pop(0)
                mx = max(mx, node.val)
                if node.left:
                    qu.append(node.left)
                if node.right:
                    qu.append(node.right)
            res.append(mx)
        return res