【二叉树】从前序与中序构造二叉树

0x00 题目

根据一棵树的 ​​前序​​​ 遍历与 ​​中序​​​ 遍历构造二叉树
注意:
你可以假设树中 ​​​没有​​ 重复的元素

---

0x01 思路

对于 ​​前序​​​ 遍历，​​第一个​​​ 元素必为​​根​​​ 只要找到 ​​第一个​​ 元素在 ​​中序​​ 遍历中的 ​​位置​​ 就可以将数组一分为二，
​​左侧​​ 为 ​​左子树​​ 的遍历结果，​​右侧​​ 为 ​​右子树​​ 对拆分后的两个数组​​递归​​的进行上述操作即可

---

0x02 解法

语言：​​Swift​​

树节点：​​TreeNode​​

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func buildTree2(_ preorder: [Int], _ inorder: [Int]) -> TreeNode? {
    var dict: [Int: Int] = [:]
    
    // 中序遍历，值对应的索引
    for i in 0..<inorder.count {
        dict[inorder[i]] = i
    }
    
    func build(_ preorder: [Int], _ preStart: Int, _ preEnd: Int, _ inorder: [Int], _ inStart: Int, _ inEnd: Int) -> TreeNode? {
        if preStart > preEnd || inStart > inEnd {
            return nil
        }
        // 根节点
        let val = preorder[preStart]
        let root = TreeNode(val)
        
        // 中序遍历中的位置
        let index = dict[val]!
        
        // 左子树长度
        let leftSize = index - inStart
        
        // 构建左子树
        root.left = build(preorder, preStart + 1, preStart + leftSize,
                          inorder, inStart, index - 1)
                          
        // 构建右子树                          
        root.right = build(preorder, preStart + leftSize + 1, preEnd,
                           inorder, index + 1, inEnd)
        return root
    }
    
    let node = build(preorder, 0, preorder.count - 1, inorder, 0, inorder.count - 1)
    
    return node
}

---