【树-中等】103. 二叉树的锯齿形层序遍历

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

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

3

/
9 20
/
15 7

返回锯齿形层序遍历如下:

[
[3],
[20,9],
[15,7]
]
【代码】
在这里插入图片描述

class Solution:
    def dfs(self,root,dep):
        if not root:
            return 
        if dep>len(self.ans):
            self.ans.append([root.val])
        else:
            self.ans[dep-1].append(root.val)
        self.dfs(root.left,dep+1)
        self.dfs(root.right,dep+1)
    def zigzagLevelOrder(self, root: TreeNode) -> List[List[int]]:
        self.ans=[]
        self.dfs(root,1)
        for index,x in enumerate(self.ans):
            if index%2:
                self.ans[index]=self.ans[index][::-1]
        return self.ans

猜你喜欢

转载自blog.csdn.net/kz_java/article/details/119801658