class TreNode {var data : Int;var left : TreNode?;var right : TreNode?;init(t : Int){self.data = t;self.left = nil;self.right = nil;}}class Solution {var indexMap = [Int:Int]()func buildTree(_ preorder: [Int], _ inorder: [Int]) -> TreNode? {// 构造哈希映射,帮助我们后续可以快速定位根节点for i in 0..<preorder.count {indexMap[inorder[i]] = i}return myBuildTree(preorder,inorder,0,preorder.count-1,0,inorder.count-1)}func myBuildTree(_ preorder: [Int], _ inorder: [Int], _ preorder_left:Int, _ preorder_right: Int,_ inorder_left:Int, _ inorder_right:Int) -> TreNode? {if preorder_left > preorder_right {return nil}// 前序遍历中的第一个节点就是根节点let preorder_root = preorder_left// 在中序遍历中定位根节点let inorder_root = indexMap[preorder[preorder_root]]!// 先把根节点建立出来let root = TreNode.init(t: preorder[preorder_root])// 得到左子树中的节点数目let size_left_subtree = inorder_root - inorder_left// 递归地构造左子树,并连接到根节点// 前序遍历中「从 左边界+1 开始的 size_left_subtree」个元素就对应了中序遍历中「从 左边界 开始到 根节点定位-1」的元素root.left = myBuildTree(preorder, inorder, preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1)// 递归地构造右子树,并连接到根节点// 前序遍历中「从 左边界+1+左子树节点数目 开始到 右边界」的元素就对应了中序遍历中「从 根节点定位+1 到 右边界」的元素root.right = myBuildTree(preorder, inorder, preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right)return root ;}}
链接: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)
