105. Construct Binary Tree from Preorder and Inorder Traversal.go

根据前序和中序遍历结果,构建二叉树
前序遍历的第一个节点,是数的根节点
从中序序列中找到根节点,则左边的都是左子树,右边的都是右子树
然后递归处理

type TreeNode struct {
    Val   int
    Left  *TreeNode
    Right *TreeNode
}

func buildTree(preorder []int, inorder []int) *TreeNode {
    if len(preorder) == 0 {
        return nil
    }
    res := &TreeNode{
        Val: preorder[0],
    }
    if len(preorder) == 1 {
        return res
    }
    idx := func(val int, nums []int) int {
        for i, v := range nums {
            if v == val {
                return i
            }
        }
        return -1
    }(res.Val, inorder)
    if idx == -1 {
        return nil
    }
    res.Left = buildTree(preorder[1:idx+1], inorder[:idx])
    res.Right = buildTree(preorder[idx+1:], inorder[idx+1:])
    return res
}
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容