【LeetCode每天一题】Binary Tree Zigzag Level Order Traversal(二叉树的之字形遍历)

Given a binary tree, return the zigzag level order traversal of its nodes' values. (ie, from left to right, then right to left for the next level and alternate between).

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

    3
   / 
  9  20
    /  
   15   7

 

return its zigzag level order traversal as:

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

思路

---

　　对于二叉树的之字形遍历，我们可以使用像之前对二叉树进行层次遍历时一样，只不过这里对每一层节点在添加子节点时的顺序有要求，当为基数层时，我们在添加节点时选择先添加左边的子节点，在添加右边的子节点。 当为偶数层时，我们应该先添加右边的子节点，在添加左边的子节点。然后遍历完毕之后在将该层节点的遍历结果添加到最终列表中。
解决代码

---

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def zigzagLevelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if not root:
            return []
        
        res, index = [], 0  # 结果列表 和  当前层数变量
        stack = [root]      # 使用辅助栈
        while stack:
            tem_stack, count = [], len(stack)     # 申请一个空列表来存储当前层的子节点， count表示当前层中的节点个数
            level, index = [], index + 1        # level 表示存储当前层的节点值的遍历顺序， 
            for _ in range(count):              # 当前层有多少个节点就遍历多少次
                tem = stack.pop()              # 弹出顶部元素
                if index % 2 == 1:              # 判断当前层，然后选择不同的子节点添加规则。
                    if tem.left:                # 奇数层先添加左节点，在添加右节点
                        tem_stack.append(tem.left)
                    if tem.right:
                        tem_stack.append(tem.right)
                else:                           # 偶数层先添加右节点在添加左节点。
                    if tem.right:
                        tem_stack.append(tem.right)
                    if tem.left:
                        tem_stack.append(tem.left)
                        
                level.append(tem.val)          #  添加当前节点的值
            stack =  tem_stack                 # 当前层遍历结束之后，将tem_stack赋值给stack.（tem_stack表示下一层子节点的排列顺序）
            res.append(level)                  # 当前层的遍历结果添加进去
        return res