【剑指Offer 4.重建二叉树】｜刷题打卡

目录

• 题目描述
• 思路分析
• AC 代码
• 总结

掘金团队号上线，助你 Offer 临门！ 点击 查看详情

一、题目描述

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

示例1

输入

[1,2,3,4,5,6,7],[3,2,4,1,6,5,7]

返回值

{1,2,5,3,4,6,7}

说明

本题目包含复杂数据结构TreeNode

二、思路分析

难度：中等

这个题考查的是树的知识点，一般遇到树的问题都会使用递归法。所以，这道题，我们今天也是利用递归算法解决。首先找到每次递归的根结点，因为前序遍历的第一个元素就是根结点。然后再分别处理左子树和右子树。

三、AC 代码

语言：Go

代码：

package main
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 * 
 * @param pre int整型一维数组 
 * @param vin int整型一维数组 
 * @return TreeNode类
*/
func reConstructBinaryTree( pre []int ,  vin []int ) *TreeNode {
    if len(pre) == 0 {
        return nil
    }
    root := &TreeNode{Val: pre[0]}
    mid := getMiddle(vin, pre[0])
    root.Left = reConstructBinaryTree(pre[1:mid+1], vin[0:mid])
    root.Right = reConstructBinaryTree(pre[mid+1:len(pre)], vin[mid+1:len(vin)])

    return root
}
// 在中序遍历中查找根结点
func getMiddle(arr []int, val int) int {
    index := 0
    for i, v := range arr {
        if v == val {
            index = i
            return index
        }
    }
    return index
}

通过截图：

四、总结

一般遇到树的问题最好优先考虑递归算法解决，本题提供一种思路，希望对大家所有帮助。