先序+中序重建二叉树

AI_Younger_Man

于 2020-06-09 09:11:42 发布

阅读量202

点赞数

分类专栏： # Leetcode

本文链接：https://blog.youkuaiyun.com/qq_38888209/article/details/106633874

版权

Leetcode 专栏收录该内容

84 篇文章

订阅专栏

根据给定的二叉树前序遍历和中序遍历序列，重建二叉树。题目提供了一组示例序列：前序{1,2,4,7,3,5,6,8}，中序{4,7,2,1,5,3,8,6}。" 104393505,9201691,将局部变量转换为全局变量的方法,"['JavaScript', '变量作用域', '前端开发', '编程技巧']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述
输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。
重建二叉树

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, tin):
        # write code here
        if not pre and not tin:
            return None
        if len(pre) != len(tin):
            return None

        root = pre[0]
        rootNode = TreeNode(root)
        root_index = tin.index(root)

        tin_left = tin[:root_index]
        tin_right = tin[root_index+1:]

        pre_left = pre[1:root_index+1]
        pre_right = pre[root_index+1:]

        left_kid = self.reConstructBinaryTree(pre_left, tin_left)
        right_kid = self.reConstructBinaryTree(pre_right, tin_right)
        
        rootNode.left = left_kid
        rootNode.right = right_kid
        return rootNode

在这里插入图片描述