题解 | #重建二叉树#

package main

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

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param preOrder int整型一维数组 
 * @param vinOrder int整型一维数组 
 * @return TreeNode类
*/
func reConstructBinaryTree( preOrder []int ,  vinOrder []int ) *TreeNode {
    // write code here

    if len(preOrder) == 0 || len(vinOrder) == 0 {
        return nil
    }

    val := preOrder[0]

    root := &TreeNode{Val:val}

    for i:=0; i < len(vinOrder); i++ {
        if vinOrder[i] == val {
            root.Left = reConstructBinaryTree(preOrder[1:i+1], vinOrder[:i])
            root.Right = reConstructBinaryTree(preOrder[i+1:], vinOrder[i+1:])
            break
        } 
    }
    return root
}