LintCode 71. Binary Tree Zigzag Level Order Traversal

Sheery_

于 2019-10-04 14:38:54 发布

阅读量166

点赞数

分类专栏：算法练习文章标签： LintCode LeetCode 二叉树遍历

本文链接：https://blog.csdn.net/qq_25590283/article/details/102071306

版权

算法练习专栏收录该内容

6 篇文章 0 订阅

订阅专栏

本文介绍了如何解决LintCode上的71题，即二叉树的锯齿形层次遍历。该问题要求在遍历二叉树时，偶数层的节点顺序从右到左，奇数层保持从左到右。解决方案利用了Python的deque，并通过跟踪层数i来判断何时反转当前层的节点顺序。提交的代码在效率上表现出色，超越了89.60%的参赛者提交的解决方案。

摘要由CSDN通过智能技术生成

题目描述：

给出一棵二叉树，返回其节点值的锯齿形层次遍历（先从左往右，下一层再从右往左，层与层之间交替进行）

分析：

此题和69. Binary Tree Level Order Traversal很像。只是需要在偶数层的时候，将此level的节点列表倒序。

需要用到python中的deque。71题代码如下：

from collections import deque
class Solution:
    def zigzagLevelOrder(self, root):
        # write your code here
        if root is None:
            return []
        else:
            que = deque([root])
            res = []
            i = 0
            while que:
                Node_perlevel = []
                i+=1
                for _ in range(len(que)):
                    node = que.popleft()
                    Node_perlevel.append(node.val)
                    if node.left:
                        que.append(node.left)
                    if node.right:
                        que.append(node.right)
                if i % 2 == 0:
                    Node_perlevel.reverse()
                    res.append(Node_perlevel)
                else:
                    res.append(Node_perlevel)
        
        return res

Your submission beats 89.60% Submissions!

可见算法效率还是不错的。

和69题code对比一下：

from collections import deque

class Solution:
    def levelOrder(self, root):
        if root is None:
            return []
        else:
            queue = deque([root])   
            res = []
            while queue:
                Node_perlevel = []   
                for _ in range(len(queue)):
                    node = queue.popleft()
                    Node_perlevel.append(node.val)
                    
                    if node.left:
                        queue.append(node.left)
                    if node.right:
                        queue.append(node.right)
                res.append(Node_perlevel)        
        return res

其实就是多了一个i的判断，i用来存储层数。