LeetCode | 面試題07. 重建二叉樹【劍指Offer】【Python】

LeetCode 面試題07. 重建二叉樹【劍指Offer】【Medium】【Python】【二叉樹】【遞歸】

問題

力扣

輸入某二叉樹的前序遍歷和中序遍歷的結果，請重建該二叉樹。假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。

例如，給出

前序遍歷 preorder = [3,9,20,15,7]
中序遍歷 inorder = [9,3,15,20,7]

返回以下的二叉樹：

3
   / \
  9  20
    /  \
   15   7

限制：

0 <= 節點個數 <= 5000

注意：本題與主站 105 題 重複

思路

遞歸

前序遍歷：根左右
中序遍歷：左根右

因而，每次取前序遍歷的值，表示根，再到中序遍歷中肯定索引。
再根據索引，分割成左子樹和右子樹。如此遞歸。

注意：
保證遞歸的 preorder 和 inorder 個數一致。

時間複雜度: O(n)，n 爲節點個數。
空間複雜度: O(n)，n 爲節點個數。

Python3代碼

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        if not preorder:
            return None
        root = TreeNode(preorder[0])

        # 根在中序遍歷中的索引
        i = inorder.index(root.val)
        # left: preorder[1] ~ preorder[i], inorder[0] ~ inorder[i-1]
        root.left = self.buildTree(preorder[1:i+1], inorder[:i])
        # right: preorder[i+1] ~ preorder[-1], inorder[i+1] ~ inorder[-1]
        root.right = self.buildTree(preorder[i+1:], inorder[i+1:])

        return root

代碼地址

GitHub連接