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

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;
}

你可能感兴趣的:(leetcode,java,算法,从中序与后序遍历序列构造二叉树)