二叉树层序遍历
问题描述:
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
分析
先写一个简单的层序遍历
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
import queue
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
arr=[]
q=queue.Queue()
q.put(root)
arr.append(root.val)
#出队
while not q.empty():
list1=[]
Node=q.get()
if Node.left:
q.put(Node.left)
arr.append(Node.left.val)
if Node.right:
q.put(Node.right)
arr.append(Node.right.val)
return arr
可以看出我们需要的结果是[[3],[9,20],[15,7]],而不是[3,9,20,15,7]
即我们需要将没一层对应的二叉树元素保存到一个数组中。那么我们可以新建一个列表来保存节点。如果列表不为空。则遍历该列表,并将这些元素保存到下一个列表中。