zoukankan      html  css  js  c++  java
  • leetcode101 二叉树的层次遍历

    这道题相当于标准DFS的一点变化。
    题目如下

    给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
    
    例如:
    给定二叉树: [3,9,20,null,null,15,7],
    
        3
       / 
      9  20
        /  
       15   7
    返回其层次遍历结果:
    
    [
      [3],
      [9,20],
      [15,7]
    ]
    

    双队列+DFS题解

    开这道题目之前,我一看,这不是标准的DFS吗,为什么会归为Medium这个难度,后来仔细一想,这个题目需要把我们通过队列的值记录下来,一开始没有想到怎么样才可以记录下这些通过的值,后来自己思考了一会,想到了之前二叉树的最大深度二叉树的最小深度这两道题的栈解法,就是使用两个栈来实现记忆数字,这样就得到了我们的题解。使用两个队列来实现节点和层数的记忆化。这样可以把时间复杂度从O(logn2)O(log n^2)变成O(n)O(n)。这次是拿python写的,因为java的队列结构操作还不是很熟。代码如下

    执行用时 : 24 ms, 在Binary Tree Level Order Traversal的Python提交中击败了100.00% 的用户
    内存消耗 : 12.1 MB, 在Binary Tree Level Order Traversal的Python提交中击败了33.50% 的用户
    
    class Solution(object):
        def levelOrder(self, root):
            """
            :type root: TreeNode
            :rtype: List[List[int]]
            """
            if root == None:
                return []
            #使用下面两个队列来辅助实现
            nodeQueue = [root]
            numberQueue = [1]
            ans = [[root.val]]
            while nodeQueue:
                node = nodeQueue.pop(0)
                number = numberQueue.pop(0)
                if node.left != None:
                    nodeQueue.append(node.left)
                    numberQueue.append(number+1)
                    try:
                        # print(node.left.val)
                        ans[number]+=[node.left.val]
                    except:
                        # print(node.left.val)
                        ans.append([node.left.val])
                if node.right != None: 
                    nodeQueue.append(node.right)
                    numberQueue.append(number+1)
                    try:
                        # print(node.right.val)
                        ans[number]+=[node.right.val]
                    except:
                        # print(node.right.val)
                        ans.append([node.right.val])
            return ans
    
  • 相关阅读:
    Discuz X3.2 分区 gid 完美伪静态方法 Apache/Nginx
    如何批量转换 WordPress 文章分类
    修改 WordPress 文件上传目录
    如何验证 jemalloc 优化 Nginx 是否生效
    .Net Core 1.1 + CentOs 7 环境配置
    DotNet Core中使用dapper
    爬虫 HttpHelper
    iTextSharp 不适用模板 代码拼接PDF
    .Net中的加密解密
    .Net Core使用 MiniProfiler 进行性能分析
  • 原文地址:https://www.cnblogs.com/yfc0818/p/11072596.html
Copyright © 2011-2022 走看看