前序与中序遍历序列构造二叉树

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
}