interview/16-LeetCode Hot 100/从前序与中序遍历序列构造二叉树.md

从前序与中序遍历序列构造二叉树

题目描述

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

解题思路

递归构造

前序遍历：[根, [左子树], [右子树]] 中序遍历：左子树], 根, [右子树

解法

func buildTree(preorder []int, inorder []int) *TreeNode {
    if len(preorder) == 0 {
        return nil
    }
    
    root := &TreeNode{Val: preorder[0]}
    index := findIndex(inorder, preorder[0])
    
    root.Left = buildTree(preorder[1:1+index], inorder[:index])
    root.Right = buildTree(preorder[1+index:], inorder[index+1:])
    
    return root
}

func findIndex(arr []int, target int) int {
    for i, v := range arr {
        if v == target {
            return i
        }
    }
    return -1
}

复杂度： O(n²) 时间（可用哈希表优化到 O(n)），O(n) 空间