题解 | #重建二叉树#

重建二叉树

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

相关推荐

嵌入式求职之路:可以看我经验😂,https://www.nowcoder.com/share/jump/73221730841876945
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务