二叉树的锯齿形层次遍历

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

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

    3
   / \
  9  20
    /  \
   15   7

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

[
  [3],
  [20,9],
  [15,7]
]

解题思路: 锯齿形层次遍历,又不单单是层次遍历,一次从左到右,一次从右向左。

先看一下二叉树的层次遍历(即逐层地从左向右访问所有节点),给定的二叉树的层次遍历结果为:

[
  [3],
  [9,20],
  [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 levelOrder(self, root):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if root is None:
            return []
        cur=[root]
        results=[]
        while cur:
            temp=[]
            val=[]
            for i in cur:
                if i.left is not None:
                    temp.append(i.left)
                if i.right is not None:
                    temp.append(i.right)
                val+=[i.val]#val数组存放每层的元素
            cur=temp
            results.append(val)
        return results

如何改进代码,使一层从左到右遍历,一层从右往左遍历,交替进行:

考虑记录树的深度,当深度为偶数时,存放每层元素的数组逆序;当深度为奇数时,存放每层元素的数组正序;

# 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):
        """
        :type root: TreeNode
        :rtype: List[List[int]]
        """
        if root is None:
            return []
        cur=[root]
        results=[]
        length=0
        while cur:
            temp=[]
            val=[]
            for i in cur:
                if i.left is not None:
                    temp.append(i.left)
                if i.right is not None:
                    temp.append(i.right)
                val+=[i.val]
            length+=1
            cur=temp
            if(length%2!=0):
                results.append(val)
            else:
                results.append(val[::-1])
                
        return results

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值