题目:
给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
解答:
这个要用队列来实现。
1. 把根节点入队列
2. 在一个循环中,先取出根节点(出队列 + 访问节点)
3. 把根节点的左右孩子入队列
然而这里不但要层次遍历,而且要按层输出,于是再增加一个list作为每一层的节点,正好队列的长度就是每一层的元素个数。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
nodeQuene = []
result = []
if not root:
return result
nodeQuene.append(root)
while nodeQuene:
# 这个表示单层节点所有的值
singleLevel = []
queneLength = len(nodeQuene)
for i in range(0, queneLength):
currentNode = nodeQuene.pop(0)
if currentNode.left:
nodeQuene.append(currentNode.left)
if currentNode.right:
nodeQuene.append(currentNode.right)
singleLevel.append(currentNode.val)
result.append(singleLevel)
return result