1. class TreNode {
    2. var data : Int;
    3. var left : TreNode?;
    4. var right : TreNode?;
    5. init(t : Int){
    6. self.data = t;
    7. self.left = nil;
    8. self.right = nil;
    9. }
    10. }
    11. class Solution {
    12. var indexMap = [Int:Int]()
    13. func buildTree(_ preorder: [Int], _ inorder: [Int]) -> TreNode? {
    14. // 构造哈希映射,帮助我们后续可以快速定位根节点
    15. for i in 0..<preorder.count {
    16. indexMap[inorder[i]] = i
    17. }
    18. return myBuildTree(preorder,inorder,0,preorder.count-1,0,inorder.count-1)
    19. }
    20. func myBuildTree(_ preorder: [Int], _ inorder: [Int], _ preorder_left:Int, _ preorder_right: Int,_ inorder_left:Int, _ inorder_right:Int) -> TreNode? {
    21. if preorder_left > preorder_right {
    22. return nil
    23. }
    24. // 前序遍历中的第一个节点就是根节点
    25. let preorder_root = preorder_left
    26. // 在中序遍历中定位根节点
    27. let inorder_root = indexMap[preorder[preorder_root]]!
    28. // 先把根节点建立出来
    29. let root = TreNode.init(t: preorder[preorder_root])
    30. // 得到左子树中的节点数目
    31. let size_left_subtree = inorder_root - inorder_left
    32. // 递归地构造左子树,并连接到根节点
    33. // 前序遍历中「从 左边界+1 开始的 size_left_subtree」个元素就对应了中序遍历中「从 左边界 开始到 根节点定位-1」的元素
    34. root.left = myBuildTree(preorder, inorder, preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1)
    35. // 递归地构造右子树,并连接到根节点
    36. // 前序遍历中「从 左边界+1+左子树节点数目 开始到 右边界」的元素就对应了中序遍历中「从 根节点定位+1 到 右边界」的元素
    37. root.right = myBuildTree(preorder, inorder, preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right)
    38. return root ;
    39. }
    40. }

    链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/solution/105-cong-qian-xu-yu-zhong-xu-bian-li-xu-7te45/
    来源:力扣(LeetCode)