当前位置: 首页 > article >正文

python-leetcode 46.从前序与中序遍历序列构造二叉树

题目:

给定两个整数preorder和inorder,其中preorder是二叉树的先序遍历,inorder是同一棵树的中序遍历,请构造二叉树并返回其根节点。


前言:

二叉树前序遍历的顺序为:先遍历根节点;随后递归地遍历左子树;最后递归地遍历右子树。

二叉树中序遍历的顺序为:先递归地遍历左子树;随后遍历根节点;最后递归地遍历右子树。

方法一:递归

对于任意一颗树而言,前序遍历的形式总是:

 [ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]

即根节点总是前序遍历中的第一个节点。而中序遍历的形式总是:

[ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]

在中序遍历中定位到根节点,那么我们就可以分别知道左子树和右子树中的节点数目。由于同一颗子树的前序遍历和中序遍历的长度显然是相同的,因此我们就可以对应到前序遍历的结果中,对上述形式中的所有左右括号进行定位。

这样以来,就知道了左子树的前序遍历和中序遍历结果,以及右子树的前序遍历和中序遍历结果,我们就可以递归地对构造出左子树和右子树,再将这两颗子树接到根节点的左右位置。

在中序遍历中对根节点进行定位时,一种简单的方法是直接扫描整个中序遍历的结果并找出根节点,但这样做的时间复杂度较高。可以考虑使用哈希表来帮助快速地定位根节点。对于哈希映射中的每个键值对,键表示一个元素(节点的值),值表示其在中序遍历中的出现位置。

在构造二叉树的过程之前,我们可以对中序遍历的列表进行一遍扫描,就可以构造出这个哈希映射。在此后构造二叉树的过程中,就只需要 O(1) 的时间对根节点进行定位了。

前序遍历:0[root[左子树][右子树]]n-1

中序遍历:0[[左子树],root,[右子树]]n-1

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: Optional[TreeNode]
        """
        #辅助递归函数,用来递归地构造二叉树,接收4个参数,表示节点区间
        def myBuildTree(preorder_left,preorder_right,inorder_left,inorder_right):
            if preorder_left>preorder_right:
                return None
            preorder_root=preorder_left#第一个节点总是当前子树的根节点
            inorder_root=index[preorder[preorder_root]]#根节点在中序遍历数组中的位置,构建的 index 哈希表可以快速找到根节点的位置。
            root=TreeNode(preorder[preorder_root])#当前子树的根节点 
            size_left_subtree=inorder_root-inorder_left#计算左子树的节点数。左子树的节点数等于中序遍历中,根节点左边部分的长度
            root.left=myBuildTree(preorder_left+1,preorder_left+size_left_subtree,inorder_left,inorder_root-1)#构造左子树。
            #左子树的节点在前序遍历中从 preorder_left + 1 到 preorder_left + size_left_subtree 之间,而在中序遍历中,节点的范围是 inorder_left 到 inorder_root - 1。递归调用 myBuildTree 来构建左子树
            root.right=myBuildTree(preorder_left+size_left_subtree+1,preorder_right,inorder_root+1,inorder_right)
            #递归构造右子树。右子树的节点在前序遍历中从 preorder_left + size_left_subtree + 1 到 preorder_right 之间,在中序遍历中从 inorder_root + 1 到 inorder_right 之间。递归调用 myBuildTree 来构建右子树。
            return root

        n=len(preorder)
        index={element:i for i,element in enumerate(inorder)}
        return myBuildTree(0,n-1,0,n-1)

时间复杂度:O(n)

空间复杂度:O(n)


方法二:迭代

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: Optional[TreeNode]
        """
        if not preorder:
                return None
        root=TreeNode(preorder[0])#创建一个树的根节点,根节点是 preorder 数组中的第一个元素
        stack=[root]
        inorderIndex=0
        for i in range(1,len(preorder)): #遍历剩余的前序遍历节点
            preorderVal=preorder[i]#获取当前遍历的前序节点值
            node=stack[-1]#取栈顶元素 node
            if node.val !=inorder[inorderIndex]: #判断当前栈顶节点的值是否与中序遍历数组中的当前元素相同,如果不同,说明当前节点是左子节点,继续构建左子树
                node.left=TreeNode(preorderVal)#如果当前节点是左子节点,就创建一个新的左子节点
                stack.append(node.left)
            else:#如果栈顶节点的值和中序遍历数组中的当前元素相同,说明当前节点是右子节点
                while stack and stack[-1].val==inorder[inorderIndex]:
                    node=stack.pop()#弹出栈顶节点,表示已经处理完它的左子树
                    inorderIndex+=1
                node.right=TreeNode(preorderVal)#创建一个新的右子节点,并将其赋给当前节点的右子树
                stack.append(node.right)
        return root

时间复杂度:O(n)

空间复杂度:O(n)

源自力扣官方题解


http://www.kler.cn/a/569420.html

相关文章:

  • 西电应用密码学与网络安全实验通关指南
  • Git与GitHub:它们是什么,有什么区别与联系?
  • coze生成的工作流,发布后,利用cmd命令行执行。可以定时发日报,周报等。让他总结你飞书里面的表格。都可以
  • C++对象特性
  • 1.Qt写简单的登录界面(c++)
  • Java---入门基础篇(下)---方法与数组
  • Python----Python爬虫(多线程,多进程,协程爬虫)
  • 运维安全计划书,驻场安全运维服务方案(Word完整版56页原件)
  • 如何在优云智算平台上面使用deepseek进行深度学习
  • C#知识|泛型Generic概念与方法
  • 【Leetcode 每日一题】132. 分割回文串 II
  • 重新审视 ChatGPT 和 Elasticsearch:第 2 部分 - UI 保持不变
  • c# winfrom增加进度条
  • WP 高级摘要插件:助力 WordPress 文章摘要精准自定义显示
  • Python Cookbook-2.13 使用C++的类iostream语法
  • 解决 Dell PowerEdge T630 增加第三方 PCIe 设备后制冷系统异常
  • 【深度学习】Hopfield网络:模拟联想记忆
  • JAVA调用Deepseek的api,完成基本对话
  • HTML AI 编程助手
  • Spring系列学习之Spring Messaging消息支持