给定两个整数数组 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;
}