LeetCode106. 从中序与后序遍历序列构造二叉树Golang版

LeetCode106. 从中序与后序遍历序列构造二叉树Golang版

1. 问题描述

2. 思路

1. postorder[len(postorder) - 1]作为root
2. 在inorder中找到i，inorder[0:i]左，inorder[i+1:]右
3. 在postorder中，postorder[:i]左，postorder[i:len(postorder)-1]右

3. 代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func buildTree(inorder []int, postorder []int) *TreeNode {
    if len(postorder) == 0 {
        return nil
    }

    root := &TreeNode {
        Val : postorder[len(postorder) - 1],
        Left : nil,
        Right : nil, 
    }

    i := 0
    for ; i < len(inorder); i++ {
        if root.Val == inorder[i] {
            break
        }
    }

    root.Left = buildTree(inorder[:i], postorder[:i])
    root.Right = buildTree(inorder[i+1:], postorder[i:len(postorder)-1])
    return root
}