【Leetcode_总结】515. 在每个树行中找最大值 - python

Q:

您需要在二叉树的每一行中找到最大的值。

示例:

输入: 

          1
         / \
        3   2
       / \   \  
      5   3   9 

输出: [1, 3, 9]

链接:https://leetcode-cn.com/problems/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]
        """
        res = []
        que = [root]
        if root == None:
            return res
        while que:
            tempList = []
            for i in range(len(que)):
                node = que.pop(0)
                tempList.append(node.val)
                if node.left:
                    que.append(node.left)
                if node.right:
                    que.append(node.right)
            res.append(max(tempList))
        return res

【Leetcode_总结】515. 在每个树行中找最大值 - python_第1张图片

你可能感兴趣的:(Leetcode,树)