leetcode - 从中序与后序遍历序列构造二叉树(106)

根据一棵树的中序遍历与后序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

参照 leetcode-从前序与中序遍历序列构造二叉树 思路

class Solution {

    private Map<Integer, Integer> map = new HashMap<>();

    public TreeNode buildTree(int[] inorder, int[] postorder) {

        for (int i = 0; i < inorder.length; i++) {
            map.put(inorder[i], i);
        }

        return deep(inorder, postorder, 0, inorder.length - 1, 0, postorder.length - 1);
    }

    private TreeNode deep(int[] inorder, int[] postorder, int ii, int ij, int pi, int pj) {
        int rootEle = postorder[pj];
        TreeNode root = new TreeNode(rootEle);
        if (pi == pj) {
            return root;
        }


        int midIndex = map.get(rootEle);
        int rightLen = ij - midIndex;

        if (midIndex != ii) {
            root.left = deep(inorder, postorder, ii, midIndex - 1, pi, pj - rightLen - 1);
        }

        if (midIndex != ij) {
            root.right  = deep(inorder, postorder, midIndex + 1, ij, pj - rightLen, pj - 1);
        }

        return root;
    }
}