实现树的广度优先遍历,每一层上的数据按照从左到右的顺序排列。
参考:链接 将树每一层的节点存在一个列表中,遍历列表中的元素,如果该节点有左右节点的话,就把它们加入一个临时列表,这样当遍历结束时,下一层的节点也按照顺序存储好了,不断循环直到下一层的列表为空。
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
tree = []
if not root:
return tree
curr_level = [root]
# print(type(root), type(curr_level)) # (<class 'precompiled.treenode.TreeNode'>, <type 'list'>)
# print(curr_level) # 作为list,却并不能遍历整个树
while curr_level:
level_list = []
next_level = []
for temp in curr_level:
level_list.append(temp.val)
if temp.left:
next_level.append(temp.left)
if temp.right:
next_level.append(temp.right)
tree.append(level_list)
curr_level = next_level
return tree
补充:
class Solution(object):
def preorder(self, root, level, res):
if root:
if len(res) < level+1: res.append([])
res[level].append(root.val)
self.preorder(root.left, level+1, res)
self.preorder(root.right, level+1, res)
def levelOrder(self, root):
res=[]
self.preorder(root, 0, res)
return res
print(temp)
得到了:<precompiled.treenode.TreeNode object at 0x7fbb9a294b50>
<precompiled.treenode.TreeNode object at 0x7fbb9a294d50>
<precompiled.treenode.TreeNode object at 0x7fbb9a294d90>
<precompiled.treenode.TreeNode object at 0x7fbb9a294dd0>
<precompiled.treenode.TreeNode object at 0x7fbb9a294e10>
说明确实里面还是树结构。 应该用这个解释来理解,廖雪峰-迭代,可以再看一遍。