【刷题】python 根据前序和中序重建二叉树(leetcode)

最新推荐文章于 2024-03-09 22:40:01 发布

小由之

最新推荐文章于 2024-03-09 22:40:01 发布

阅读量902

点赞数 2

分类专栏：数据结构 python 二叉树

本文链接：https://blog.csdn.net/weixin_42486139/article/details/106722873

版权

python 同时被 3 个专栏收录

7 篇文章 0 订阅

订阅专栏

数据结构

1 篇文章 0 订阅

订阅专栏

二叉树

1 篇文章 0 订阅

订阅专栏

本文介绍了一种从给定的前序和中序序列重构二叉树的方法，并通过层序遍历验证了二叉树的正确性。首先，从前序序列中找到根节点，然后在中序序列中定位根节点，以此区分左右子树。接着，递归地应用这一过程直至所有节点都被处理。最后，使用层序遍历检查重构的二叉树。

摘要由CSDN通过智能技术生成

步骤：

1.在前序序列中找到根节点

2.在中序序列中，根据根节点的位置，区分左右子树集合

3.分别列出左右子树的前序序列和中序序列

4.回溯前序序列，重复上述步骤，直到将所有节点集合拆解为叶子节点

举例：

前序（根-左-右）：1 2 4 7 3 5 6 8

中序（左-根-右）：4 7 2 1 5 3 8 6

1.前序第一个数字是整个二叉树的根节点，根节点：1

2.中序序列，以根节点1区分左右子树集合，左子树集合：4 7 2，右子树集合：5 3 8 6

3.左子树的前序序列：2 4 7，中序序列：4 7 2。右子树的前序序列：3 5 6 8 ，中序序列： 5 3 8 6

4.根据拆分的前序序列和中序序列，继续确定根节点，继续拆分

代码：

定义二叉树的类

class TreeNode:
    def __init__(self,x):
        self.val = x
        self.left = None
        self.right = None

递归重建二叉树

class Solution:
    def reConstructBinaryTree(self, pre, tin):#pre、tin分别是前序序列和中序序列
        if len(pre)>0:
            root = TreeNode(pre[0]) #前序序列的第一个肯定是当前子树的根节点
            rootidx = tin.index(root.val) #返回根节点在中序序列中的位置
            root.left = self.reConstructBinaryTree(pre[1:1+rootidx],tin[:rootidx])#重建左子树
            root.right = self.reConstructBinaryTree(pre[1+rootidx:],tin[rootidx+1:])#重建右子树
            return root

可以采用层序输出的方式验证生成的二叉树是否正确，用先进先出的队列依次保存层序遍历到的节点（该方法在类Solution下），然后遍历每个节点的左子节点和右子节点，代码实现如下：

    def PrintFromTopToBottom(self, root):
        ans=[]
        if root==None:
            return ans
        else:
            q=[root]
            while q:
                node=q.pop(0)
                ans.append(node.val)
                if node.left:
                    q.append(node.left)
                if node.right:
                    q.append(node.right)
            return ans

完整代码：

class TreeNode:
    def __init__(self,x):
        self.val = x
        self.left = None
        self.right = None
        
class Solution:
    def reConstructBinaryTree(self, pre, tin):#pre、tin分别是前序序列和中序序列
        if len(pre)>0:
            root = TreeNode(pre[0])
            rootidx = tin.index(root.val)
            root.left = self.reConstructBinaryTree(pre[1:1+rootidx],tin[:rootidx])
            root.right = self.reConstructBinaryTree(pre[1+rootidx:],tin[rootidx+1:])
            return root
    
    def PrintFromTopToBottom(self, root):
        ans=[]
        if root==None:
            return ans
        else:
            q=[root]
            while q:
                node=q.pop(0)
                ans.append(node.val)
                if node.left:
                    q.append(node.left)
                if node.right:
                    q.append(node.right)
            return ans
        
pre = [1, 2, 4, 7, 3, 5, 6, 8]
tin = [4, 7, 2, 1, 5, 3, 8, 6]
a = Solution()
root = a.reConstructBinaryTree(pre,tin)
ans = a.PrintFromTopToBottom(root)
print(ans)

输出： [1, 2, 3, 4, 5, 6, 7, 8]