呵呵呵呵这个题真的花了很久才懂!
题目给了两个数组,分别是:
inorder = [9,3,15,20,7](3左边是左子树,右边是右子树;“20”同理:15是左子树,7是右子树。)
postorder = [9,15,7,20,3] (左子树 右子树 根结点)
要我们建立一棵这样的树。
在做所有的事情之前,首先创建一个hashmap,存inorder里面的每个元素的index。
规律:
第一,postorder的最后一个元素是树的根结点
所以:
TreeNode root = new TreeNode (postorder[poEnd]);
得到每个子树的root在inOrder里面的index之后,把它记下来(poEnd就是每个root在postorder里面的index):
int inRoot = map.get(postorder[poEnd]);
第二,寻找left child:
因为inRoot就是这个根结点在inorder里面的index,所以inEnd(第二个parameter)应该更新成 inRoot -1(找左子树就往左边去一个;D)。
同理,在postOrder里面,poStart保持不变,poEnd变成poStart加上(inRoot - inStart - 1):也就是左子树在postorder里面的范围。
root.left = helper (inStart, inRoot -1, poStart, poStart + (inRoot - inStart - 1),
inorder, postorder, map);
第三,寻找right child。
与上面同理,在inorder里面往右边移(inRoot+1),inEnd保持不变,在原有的poStart的基础上加上inRoot-inStart,poEnd-1来排除现在我们所在的这个root。
root.right = helper(inRoot + 1, inEnd, poStart + (inRoot - inStart), poEnd - 1,
inorder, postorder, map);
class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
HashMap<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < inorder.length; i++){
map.put(inorder[i], i);
}
return helper(0, inorder.length-1, 0, postorder.length-1, inorder, postorder, map);
}
private TreeNode helper (int inStart, int inEnd, int poStart, int poEnd, int [] inorder,
int [] postorder, HashMap<Integer, Integer> map){
if(inStart > inEnd || poStart > poEnd){
return null;
}
TreeNode root = new TreeNode (postorder[poEnd]);
int inRoot = map.get(postorder[poEnd]); //Get the index of postorder[poEnd] in inorder
root.left = helper (inStart, inRoot -1, poStart, poStart + (inRoot - inStart - 1), inorder, postorder, map);
root.right = helper(inRoot + 1, inEnd, poStart + (inRoot - inStart), poEnd - 1, inorder, postorder, map);
return root;
}
}
Happy LeetCoding ?