二叉树的层次遍历-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
全部评论

相关推荐

Lorn的意义:你这种岗位在中国现在要么牛马天天加班,要么关系户进去好吃好喝,8年时间,真的天翻地覆了,对于资本来说你就说一头体力更好的牛马,哎,退伍没有包分配你真的亏了。
点赞 评论 收藏
分享
流浪的神仙:无恶意,算法一般好像都得9硕才能干算法太卷啦
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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