LeetCode Binary Tree Level Order Traversal

来源:互联网 发布:淘宝情趣用品保密吗 编辑:程序博客网 时间:2024/06/16 05:46

LeetCode解题之Binary Tree Level Order Traversal


原题

实现树的广度优先遍历,每一层上的数据按照从左到右的顺序排列。

注意点:

例子:

输入:

    3   / \  9  20    /  \   15   7

输出:

[  [3],  [9,20],  [15,7]]

解题思路

将树每一层的节点存在一个列表中,遍历列表中的元素,如果该节点有左右节点的话,就把它们加入一个临时列表,这样当遍历结束时,下一层的节点也按照顺序存储好了,不断循环直到下一层的列表为空。

AC源码

# Definition for a binary tree node.class TreeNode(object):    def __init__(self, x):        self.val = x        self.left = None        self.right = Noneclass Solution(object):    def levelOrder(self, root):        """        :type root: TreeNode        :rtype: List[List[int]]        """        result = []        if not root:            return result        curr_level = [root]        while curr_level:            level_result = []            next_level = []            for temp in curr_level:                level_result.append(temp.val)                if temp.left:                    next_level.append(temp.left)                if temp.right:                    next_level.append(temp.right)            result.append(level_result)            curr_level = next_level        return resultif __name__ == "__main__":    None

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。

0 0