Python描述 LeetCode 515. 在每個樹行中找最大值

大家好，我是亓官劼（qí guān jié ），在【亓官劼】公眾號、CSDN、GitHub、B站等平台分享一些技術博文，主要包括前端開發、python後端開發、小程序開發、數據結構與算法、docker、Linux常用運維、NLP等相關技術博文，時光荏苒，未來可期，加油~

如果喜歡博主的文章可以關注博主的個人公眾號【亓官劼】(qí guān jié)，裡面的文章更全更新更快。如果有需要找博主的話可以在公眾號後台留言，我會盡快回復消息.

---

本文原創為【亓官劼】（qí guān jié ），請大家支持原創，部分平台一直在惡意盜取博主的文章！！！ 全部文章請關注微信公眾號【亓官劼】。

題目

給定一棵二叉樹的根節點 root ，請找出該二叉樹中每一層的最大值。

示例1：

輸入: root = [1,3,2,5,3,null,9]
輸出: [1,3,9]

示例2：

輸入: root = [1,2,3]
輸出: [1,3]

提示：

• 二叉樹的節點個數的范圍是 [0,104]
• -231 <= Node.val <= 231 - 1

Python描述

層次遍歷

# 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 not root:
return []
res = []
q = [root]
while q:
tmp_max,n = q[0].val,len(q)
for i in range(n):
node = q[0]
tmp_max = max(node.val,tmp_max)
q.remove(q[0])
if node.left:
q.append(node.left)
if node.right:
q.append(node.right)
res.append(tmp_max)
return res