python-leetcode103-树的宽度遍历二叉树的锯齿形层次遍历

Posted 2020-11-27 西西嘛呦

给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

3
/
9 20
    /
  15 7
返回锯齿形层次遍历如下：

[
[3],
[20,9],
[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 zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
        from collections import deque
        if not root:
            return []
        queue=deque()
        queue.append(root)
        res=[]
        level=0
        while queue:
            tmp=deque()
            for i in range(len(queue)):
                t=queue.popleft()
                if level%2==0:
                    tmp.append(t.val)                
                else:
                    tmp.appendleft(t.val)
                if t.left:
                    queue.append(t.left)
                if t.right:
                    queue.append(t.right)
            level+=1
            res.append(list(tmp))
        return res

网上那些python版本的都是先层次遍历，然后在对奇数层进行翻转再加入到结果中，其实大可不必。