请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7],
# 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: TreeNode) -> List[List[int]]:
if not root:
return []
stack, res = [root], []
layer = 0
while stack:
layer+=1
#lay代表下一层,layval存储当前层的值
next_lay, layval = [], []
for node in stack:
#存储当前层的值,由于当前层已经是按照从左到右或者从右到左拍好,所以直接取值就满足题目要求
layval.append(node.val)
if node.left:
next_lay.append(node.left)
if node.right:
next_lay.append(node.right)
stack= next_lay
#res.append(layval)
if layer%2!=0:
res.append(layval)
else:
res.append(layval[::-1])
return res