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

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

例如:
给定二叉树 [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版本的都是先层次遍历,然后在对奇数层进行翻转再加入到结果中,其实大可不必。

猜你喜欢

转载自www.cnblogs.com/xiximayou/p/12392923.html
今日推荐