【剑指 Offer】面试题32 - I. 从上到下打印二叉树

原创已于 2022-03-02 11:09:05 修改 · 142 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #链表 #动态规划

于 2022-03-02 11:09:03 首次发布

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

本文探讨了两种不同的二叉树层次遍历的Python实现方法。第一种方法使用基本的队列操作，而第二种方法利用了deque（双端队列）来提高效率。两种方法都在给定的测试用例中成功通过，但第二种方法在执行时间和内存消耗上表现更优，分别击败了83.81%和36.22%的用户。

题目
代码
执行用时：40 ms, 在所有 Python3 提交中击败了32.56% 的用户
内存消耗：15.4 MB, 在所有 Python3 提交中击败了7.48% 的用户
通过测试用例：34 / 34

class Solution:
    def levelOrder(self, root: TreeNode) -> List[int]:
        if root is None:
            return []
        queue=[root]
        ans=[]
        while queue:
            temp=queue.pop(0)
            ans.append(temp.val)
            if temp.left:
                queue.append(temp.left)
            if temp.right:
                queue.append(temp.right)
        return ans

【方法2】采用更优秀的api
执行用时：32 ms, 在所有 Python3 提交中击败了83.81% 的用户
内存消耗：15.3 MB, 在所有 Python3 提交中击败了36.22% 的用户
通过测试用例：34 / 34

class Solution:
    def levelOrder(self, root: TreeNode) -> List[int]:
        if not root: return []
        res, queue = [], collections.deque()
        queue.append(root)
        while queue:
            node = queue.popleft()
            res.append(node.val)
            if node.left: queue.append(node.left)
            if node.right: queue.append(node.right)
        return res