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

题目:

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

分析:

前序序列是按照「根左右」的顺序遍历,而中序序列是按照「左根右」的顺序遍历,所以根节点肯定是前序序列中的第一个元素,然后再中序序列中找到该根节点的位置,将其分为左右子树,递归构建。

// date 2023/10/23
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func buildTree(preorder []int, inorder []int) *TreeNode {
    if len(preorder) == 0 {
        return nil
    }
    root := &TreeNode{Val: preorder[0]}
    index := 0
    for i := 0; i < len(inorder); i++ {
        if inorder[i] == preorder[0] {
            index = i
            break
        }
    }
    root.Left = buildTree(preorder[1:index+1], inorder[0:index])
    root.Right = buildTree(preorder[index+1:], inorder[index+1:])
    return root
}

最后更新于