题解 | #按之字形顺序打印二叉树#

按之字形顺序打印二叉树

https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 
# @param pRoot TreeNode类 
# @return int整型二维数组
# 此题为非常经典的双端队列的应用
class Solution:
    def Print(self , pRoot: TreeNode) -> List[List[int]]:
        # write code here
        from collections import deque
        if not pRoot:
            return []
        res = []
        DQ = deque([pRoot])
        level = 1 # 控制左右交替运行
        length = 1 
        while DQ: # 正常流程队列的层次遍历
            level_res = []
            for _ in range(length):
                if level % 2 == 0: # 要点在这,如果从左侧出队,那么左右子从右侧入队,顺序为先右后左
                    cur = DQ.popleft()
                    if cur.right:
                        DQ.append(cur.right)
                    if cur.left:
                        DQ.append(cur.left)
                else:# 如果从右侧出队,则左右子就要从左侧入队,顺序为先左后右
                    cur = DQ.pop()
                    if cur.left:
                        DQ.appendleft(cur.left)
                    if cur.right:
                        DQ.appendleft(cur.right)

                level_res.append(cur.val) # 存层次
            level += 1 # 左右交替
            res.append(level_res)
            length = len(DQ)  
        return res

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-04 18:02
好不容易拿到了字节Offer,鼠鼠做后端的,但家里人觉得可能被裁员不稳定,让鼠鼠去投国企,现在好纠结到底该咋选
文档传偷助手:该投就投吧,不过建议别放弃offer 拿到手里的才是最好的
投递字节跳动等公司9个岗位
点赞 评论 收藏
分享
仁者伍敌:牛子这些人还会点一个自动回复,boss都不带回复的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务