leetcode —— 面试题32 - III. 从上到下打印二叉树 III

题目:请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。

——————————————————

解题思路:使用广度优先遍历,但是题目要求每一层的输出顺序和其上一层或下一层的输出顺序是相反的。因此在广度优先遍历中将结点放进队列中的时候,我们同时保存该结点所在的层数,即放进队列中的元素是[node,d]。

其中node表示树的节点,d表示该结点是第几层的节点,默认根节点的层数为0,越往下层数越高。

当层数是偶数层时,我们直接使用append将节点的数据存进列表中。

当层数是奇数时,我们使用insert将节点的数据存进列表中,这样就实现了将数据逆序存放。

# 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: TreeNode) -> List[List[int]]:
        if not root:
            return []
            
        queue = collections.deque([[root,0]])  # 创建队列,保存节点值和节点所在层数
        nums = []  # 用于保存整个遍历的值
        
        while queue:
            num = []  # 用于保存每层的遍历结果
            
            for _ in range(len(queue)):  # 对每一层进行遍历
                node,d = queue.popleft()
                
                if d%2 == 0:  # 如果当前层数为偶数层,则顺序存放
                    num.append(node.val)
                else:    # 如果当前层数为奇数层,则逆序存放
                    num.insert(0,node.val)
                    
                if node.left: 
                    queue.append([node.left,d+1])
                if node.right:
                    queue.append([node.right,d+1])
            nums.append(num[:])
        return nums

该方法中的insert()的时间复杂度为O(n),但是整体的时间复杂度应该也还是O(n)。空间复杂度为O(n)。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值