二叉树的层次遍历-Python 两种写法(列表生成式)

1、扫描一遍 一边生成遍历结果一边给deque添加后继节点

def Print(self, pRoot):
        # write code here
        if not pRoot:
            return []
        ans=[]
        deque=[pRoot]
        while deque:
             length=len(deque)
             tmp=[]
             for _ in range(length):
                 curr=deque.pop(0)
                 tmp.append(curr.val)
                 if curr.left:deque.append(curr.left)
                 if curr.right:deque.append(curr.right)
             ans.append(tmp)
        return ans

2、扫描两遍,生成和添加分开处理

def Print(self, pRoot):
        # write code here
        if not pRoot:
            return []
        ans=[]
        deque=[pRoot]
        while deque:
              layer=deque
              ans.append([x.val for x in deque])
              deque=[x for l in layer for x in [l.left,l.right] if x]

        return ans
全部评论

相关推荐

昨天 00:37
已编辑
山东大学 C++
小浪_Coding:你问别人,本来就是有求于人,别人肯定没有义务免费回答你丫, 有点流量每天私信可能都十几,几十条的,大家都有工作和自己的事情, 付费也是正常的, 就像你请别人搭把手, 总得给人家买瓶水喝吧
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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