leetcode 105. 从前序与中序遍历序列构造二叉树
- 前序 根-左子树-右子树
- 中序 左-根-右
- 前序序列中,第一个是根,然后是所有左子树节点,再是右子树节点
- 中序序列中,左子树都在根左边,右子树都在根右边。
- 根据中序序列的左子树长度,得到右子树根节点在前序序列的位置
func buildTree(preorder []int, inorder []int) *TreeNode {
n := len(preorder)
if n == 0 {
return nil
}
// 找到根在中序的位置
var i int
for i = 0; i < n; i++ {
if inorder[i] == preorder[0] {
break
}
}
root := &TreeNode{
Val: preorder[0],
Left: buildTree(preorder[1:1+i], inorder[:i]),
Right: buildTree(preorder[i+1:], inorder[i+1:]),
}
return root
}