题解 | #重建二叉树#

重建二叉树

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:]), 
    }
}
全部评论

相关推荐

06-15 18:44
黄淮学院 Java
Lynn012:如果是居民楼还是算了吧,看着有点野呢
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务