LeetCode Java刷题笔记—106. 从中序与后序遍历序列构造二叉树

Posted 2022-12-05 刘Java

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

给定两个整数数组 inorder 和 postorder ，其中 inorder 是二叉树的中序遍历， postorder 是同一棵树的后序遍历，请你构造并返回这颗 二叉树 。

中等难度，这道题和LeetCode Java刷题笔记—105. 从前序与中序遍历序列构造二叉树非常相似，解题思路也是一样的，采用分治算法即可。

HashMap<Integer, Integer> inMap = new HashMap<>();

public TreeNode buildTree(int[] inorder, int[] postorder) 
    int len = postorder.length;
    for (int i = 0; i < len; i++) 
        inMap.put(inorder[i], i);
    
    return buildTree(postorder, 0, len - 1, 0);


TreeNode buildTree(int[] postorder, int postLeft, int postRight, int inLeft) 
    if (postLeft > postRight) 
        return null;
    
    int inRootIndex = inMap.get(postorder[postRight]);
    TreeNode root = new TreeNode(postorder[postRight]);
    int len = inRootIndex - inLeft;
    root.left = buildTree(postorder, postLeft, postLeft + len - 1, inLeft);
    root.right = buildTree(postorder, postLeft + len, postRight - 1, inRootIndex + 1);
    return root;