/\*\*
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
if(postorder == null || inorder == null || postorder.length == 0 || inorder.length == 0 || postorder.length != inorder.length\) {
return null;
}
// use hashmap to store the inorder <value,index> pair
Map<Integer,Integer> map = new HashMap<Integer,Integer>();
for (int i = 0; i < inorder.length; i++) {
map.put(inorder[i],i);
}
return helper(inorder, 0, inorder.length - 1, postorder, 0, postorder.length - 1, map);
}
private TreeNode helper(int[] in, int inleft, int inright, int[] post, int postleft, int postright,Map<Integer,Integer> map) {
// base case
if(inleft > inright) {
return null;
}
TreeNode root = new TreeNode(post[postright]);
int size = map.get(root.val) - inleft;
root.left = helper(in, inleft, inleft + size - 1 , post, postleft, postleft + size - 1, map);
root.right = helper(in, inleft + size + 1, inright , post, postleft + size, postright - 1, map);
return root;
}
}