题解 | #重建二叉树#
重建二叉树
http://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6
简易递归实现二叉树重建(先、中)
func reConstructBinaryTree( pre []int , vin []int ) *TreeNode {
// write code here
// 利用先序确定根节点位置, 再通过中序节点找到根节点, 确定左子树、右子树节点长度, 进行分治 还原节点
if len(pre) == 0 { return nil }
r, idx := pre[0], 0
// 找到
for _, v := range vin {
if v == r {
break
}
idx++
}
return &TreeNode{
r,
// 左子树
reConstructBinaryTree(pre[1:idx + 1], vin[:idx]),
// 右子树
reConstructBinaryTree(pre[idx+1:], vin[idx+1:]),
}
}