LeetCode 515. 在每个树行中找最大值
515. 在每个树行中找最大值
Difficulty: 中等
您需要在二叉树的每一行中找到最大的值。
示例:
输入:
1
/ \
3 2
/ \ \
5 3 9
输出: [1, 3, 9]
Solution
Language: ****
BFS+queue实现层序遍历,十分easy,一次AC。
# 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: TreeNode) -> List[int]:
if not root: return None
queue, res = [root], []
while queue:
curLevel, size = [], len(queue)
for i in range(size):
node = queue.pop(0)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
curLevel.append(node.val)
res.append(max(curLevel))
return res