leetcode_103. 二叉树的锯齿形层次遍历

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

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

例如:
给定二叉树 [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:
        res = []

        def dfs(root, depth):
            if root is None:
                return
            if depth == len(res):
                res.append([root.val])
            else:
                if depth % 2 == 1:
                    depth_list = res[depth]
                    temp = [root.val]
                    temp.extend(depth_list)
                    res[depth] = temp
                else:
                    res[depth].append(root.val)
            dfs(root.left, depth + 1)
            dfs(root.right, depth + 1)

        dfs(root, 0)
        return res


if __name__ == '__main__':
    # a = TreeNode(3)
    # a.left = TreeNode(9)
    # a.right = TreeNode(20)
    # a.right.left = TreeNode(15)
    # a.right.right = TreeNode(7)

    a = TreeNode(1)
    a.left = TreeNode(2)
    a.right = TreeNode(3)
    a.left.left = TreeNode(4)
    a.left.right = TreeNode(5)

    s = Solution()
    ans = s.zigzagLevelOrder(a)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 9.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值