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

相关推荐

01-15 19:59
中山大学 C++
牛客60887332...:你这是人写出来的? 本科标到硕士后面 留那么多空给 hr 填?
点赞 评论 收藏
分享
2025-12-15 11:27
门头沟学院 Java
哇哇的菜鸡oc:所有人不要理会,就好了,后面他就知道怎么回事了,只能说有的时候市场都是被宰的人搞坏的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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