leetcode 103. 二叉樹的鋸齒形層序遍歷 BFS方法

Stig.Huang發表於2020-12-23

102題為本題的“先導題”,102題解答中的表述本題解答不再贅述
https://blog.csdn.net/weixin_50791900/article/details/111569831
在102題的基礎上,僅需考慮逆序遍歷的層
在此我運用了level變數記錄該層是否需要倒序遍歷
初始設為0,每遍歷一層加一(同樣是記錄二叉樹的深度)
奇數層需要逆序
原來的遍歷順序無需改變
僅需要在逆序便利的層中將值在列表頭插入
在這裡我運用的insert方法在0位插入實現的逆序插入的效果

class Solution:
    def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return[]

        ans = []
        a = collections.deque()
        a.append(root)
        level = 0

        while a:
            n = len(a)
            b = []

            for i in range(n):
                node = a.popleft()
                if level % 2 == 0:
                    b.append(node.val)
                if level % 2 != 0:
                    b.insert(0, node.val)
                if node.left:
                    a.append(node.left)
                if node.right:
                    a.append(node.right) 

            level += 1
            ans.append(b)
        return ans

效果尚可

相關文章