LeetCode103. Binary Tree Zigzag Level Order Traversal （思路及python解法）

最新推荐文章于 2021-02-05 11:10:39 发布

原创最新推荐文章于 2021-02-05 11:10:39 发布 · 258 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

179 篇文章

订阅专栏

Python

173 篇文章

订阅专栏

本文介绍了一种二叉树的锯齿形层序遍历算法，通过BFS方法实现从左到右再从右到左交替进行的节点值遍历。以示例二叉树为基准，详细解析了如何获取锯齿形遍历结果。

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]
]

跟层序遍历差不多，只是隔一行倒过来一次。

还是使用BFS方法，用tempnode存储每一层的节点，用forward来判断这层list是正还是倒着，final存储最终结果。

思路和方法都是比较简单的。

class Solution:
    def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root: return root
        bfs=[root]
        final=[]
        forward=True
        
        while bfs:
            tempnode=[]
            tempval=[node.val for node in bfs]
            if forward:
                final.append(tempval)
            else:
                final.append(tempval[::-1])
                
            for node in bfs:
                if node.left:
                    tempnode.append(node.left)
                if node.right:
                    tempnode.append(node.right)
            
            forward = not forward
            bfs=tempnode
        
        return final