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)