【golang必备算法】二叉树构造篇

二叉树构造篇

889. 根据前序和后序遍历构造二叉树

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

func buildTree(preorder []int, inorder []int) *TreeNode {
    if len(preorder)==0 || len(inorder)==0{
        return nil
    }
    index := find(preorder,inorder) //找到分界点
    return &TreeNode{
        Val : preorder[0],
        Left : buildTree(preorder[1:len(inorder[:index])+1],inorder[:index]),
        Right : buildTree(preorder[len(inorder[:index])+1:],inorder[index+1:]),
    }
}

func find(preorder,inorder []int)int{
    for i:=0;i<len(inorder);i++{
        if preorder[0]==inorder[i]{
            return i
        }
    }
    return -1
}

关键代码解释：
第8行：

 Left : buildTree(preorder[1:len(inorder[:index])+1],inorder[:index]),

• 中序遍历直接从index分，而前序遍历可以根据中序遍历长度推断出来，因为同一个的树前序、中序数量是一样的。
• 这里加1是因为第一个是根节点，左子树的前序遍历不包括这个所以加1

第9行：

 Right : buildTree(preorder[len(inorder[:index])+1:],inorder[index+1:])，

• 中序遍历从index分，前序遍历结果是根节点、左子树递归遍历、右子树递归遍历
• preorder[:len(inorder[:index])+1]是根节点加左子树递归遍历结果
• 所以右子树前序遍历就是preorder[len(inorder[:index])+1:]

106. 从中序与后序遍历序列构造二叉树

与前序和后序遍历构造二叉树类似，都会用到一个很重的点，就是中序数组大小一定是和后序数组的大小相同的

func buildTree(inorder []int, postorder []int) *TreeNode {
    if len(inorder)==0 || len(postorder)==0{
        return nil
    }
    index := find(inorder,postorder)
    return &TreeNode{
        Val : postorder[len(postorder)-1],
        Left :buildTree(inorder[:index],postorder[:index]),
        // 同一个的树后序、中序数量是一样的
        Right :buildTree(inorder[index+1:],postorder[index:len(postorder)-1]),
    }
}

func find(inorder []int,postorder []int)int{
    for i:=0;i<len(inorder);i++{
        if inorder[i] == postorder[len(postorder)-1]{
            return i
        }
    }
    return -1
}

第9行 Right 这里：

 Right :buildTree(inorder[index+1:],postorder[index:len(postorder)-1]),

• 首先后序遍历数组postorder最后一个元素不要，它是根节点，还剩下左子树后序遍历数组+右子树后序遍历数组
• 其次因为中序数组大小一定是和后序数组的大小相同，左子树中序遍历数组为inorder[:index]
• 所以右子树后序遍历数组为postorder[index:len(postorder)-1])

参考：

代码随想录

labuladong