美文网首页
leetcode 102 二叉树层次遍历

leetcode 102 二叉树层次遍历

作者: Arsenal4ever | 来源:发表于2020-01-23 22:46 被阅读0次

以前写过,在写一遍,代码更加精简!

# 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 levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:
            return []
        level, nextLevel = [root], []
        result = []
        while level:
            result.append([i.val for i in level])
            nextLevel = []
            for node in level:
                if node.left:
                    nextLevel.append(node.left)
                if node.right:
                    nextLevel.append(node.right)
            level = nextLevel
        return result

相关文章

网友评论

      本文标题:leetcode 102 二叉树层次遍历

      本文链接:https://www.haomeiwen.com/subject/hyvazctx.html