刷题笔记day20-二叉树

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

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */

func buildTree(inorder []int, postorder []int) *TreeNode {
    // 思路1：思路采用递归的方式进行。

    if len(inorder) == 0 || len(postorder) == 0 {
        return nil
    }

    return traversal(inorder, postorder)
}

func traversal(inorder []int, postorder []int) *TreeNode {
    // 思路1：思路采用递归的方式进行。

    if len(postorder) == 0 {
        return nil
    }
    
    // 先找到后序遍历的根节点。
    rootVal := postorder[len(postorder) - 1]
    root := &TreeNode{Val: rootVal}

    // 找到根在中序遍历的位置
    rootIndex := 0
    for i, v := range inorder {
        if v == rootVal {
            rootIndex = i
        }
    }
    // 切分中序
    leftInOrder := inorder[:rootIndex]
    rightInOrder := inorder[rootIndex+1:]
    // 和后序序列
    leftPostOrder := postorder[:len(leftInOrder)]
    rightPostOrder := postorder[len(leftInOrder): len(postorder)-1]

    // 构建二叉树
    root.Left = traversal(leftInOrder, leftPostOrder)
    root.Right = traversal(rightInOrder, rightPostOrder)

    return root
}