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

根據(jù)前序和中序遍歷結(jié)果,構(gòu)建二叉樹
前序遍歷的第一個(gè)節(jié)點(diǎn),是數(shù)的根節(jié)點(diǎn)
從中序序列中找到根節(jié)點(diǎn),則左邊的都是左子樹,右邊的都是右子樹
然后遞歸處理

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
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容