题解 | #按之字形顺序打印二叉树#
按之字形顺序打印二叉树
https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0
层次遍历,只需考虑每一层的结果出来之后,是从左到右的顺序添加,还是从右到左的顺序添加。可以用一个Flag标志,也可以根据结果集的长度的奇偶性标志。
class Solution: def Print(self, pRoot: TreeNode) -> List[List[int]]: queue = [] if not pRoot: return [] queue.append(pRoot) res = [] while queue: temp = [] # temp用于存储该层的结果 for _ in range(len(queue)): c = queue.pop(0) temp.append(c.val) if c.left: queue.append(c.left) if c.right: queue.append(c.right) # 根据结果集的长度决定,顺序添加还是逆序添加。长度为0的时候顺序添加,为1的时候逆序添加。故如下。 res.append(temp[::-1] if len(res) % 2 else temp) return res