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
}