牛客刷题-重建二叉树(GO语言版)
作者:互联网
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。 解法:func reConstructBinaryTree( pre []int , vin []int ) *TreeNode { // write code here if len(pre) == 1{ return &TreeNode{Val:pre[0]} } head := TreeNode{Val:pre[0]} idx := 0 for k := range vin{ if vin[k] == pre[0]{ idx = k break } } length := len(vin[:idx]) if idx == 0{ head.Left = nil }else{ head.Left = reConstructBinaryTree(pre[1:length+1],vin[:idx]) } if length + 1 < len(vin){ head.Right = reConstructBinaryTree(pre[length+1:],vin[idx+1:]) }else{ head.Right = nil } return &head }
标签:pre,head,遍历,idx,vin,牛客,二叉树,GO 来源: https://www.cnblogs.com/lvpengbo/p/14408272.html