Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree [3,9,20,null,null,15,7],

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。
    3
   / \
  9  20
    /  \
   15   7

 

return its level order traversal as:

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

思路
  这道题可以使用广度优先遍历的方法来解决,但是相对于广度优先这里处理的情况稍微也要复杂一些,因为每一层遍历的结果都需要单独的进行使用一个列表进行存在。根据这个我们可以设置一个变量来记录每一层中的节点数,进行遍历(当然需要使用到列表这个辅助空间进行遍历)。
解决代码
  
 1 # Definition for a binary tree node.
 2 # class TreeNode(object):
 3 # def __init__(self, x):
 4 # self.val = x
 5 # self.left = None
 6 # self.right = None
 7 
 8 class Solution(object):  9     def levelOrder(self, root): 10         """
11  :type root: TreeNode 12  :rtype: List[List[int]] 13         """
14         if not root: 15             return [] 16         res = [] # 结果列表 17         stack  = [root] # 栈存当前层中的节点 18         while stack: # 当栈为空时,说明遍历结束 19             tem, count = [], len(stack) # tem存储当前层次的节点值,count存储当前层次的节点数。 20             for _ in range(count): 遍历count次 21                 node = stack.pop(0) # 从列表中第一个节点进行弹出, 并判断左右节点是否为空,进行相应的添加操作。 22                 if node.left: 23  stack.append(node.left) 24                 if node.right: 25  stack.append(node.right) 26  tem.append(node.val) # 将当前层所有节点值记录到结果中中 27  res.append(tem) 28         return res     
扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄