注意不变量为左闭右开区间
class Solution {
Map<Integer, Integer> map;
//inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
public TreeNode buildTree(int[] inorder, int[] postorder) {
map = new HashMap<>();
for (int i = 0; i < inorder.length; i++) {// 用map保存中序序列的数值对应位置
map.put(inorder[i], i);
}
return findNode(inorder, 0, inorder.length, postorder, 0, postorder.length);
}
public TreeNode findNode(int[] inorder, int inbegin, int inend, int[] postorder, int postBegin, int postend) {
if (inbegin >= inend || postBegin >= postend) {//说明不满足左闭右开,数组为空返回null
return null;
}
//第一次调用 根节点3 rootindex=1
int rootindex = map.get(postorder[postend - 1]);//找到后续遍历根节点的值,在找中序遍历中的位置
TreeNode treeNode = new TreeNode(inorder[rootindex]);
//第一次调用 中序做数组长度1-0=1 [9]
int leftLen = rootindex - inbegin;//保存中序遍历左子树长度,用来确定后续数组的长度
//左闭右开
//左序数组需要传中序左数组跟后续左数组
//其中中序切割时传入inbegin,rootindex(中序遍历此次迭代根节点位置)[0,1)[9]
//后续遍历根据leftLen左子树长度切割传入 postBegin,postBegin+leftLen[0,1)数组[9]
treeNode.left = findNode(inorder, inbegin, rootindex, postorder, postBegin, postBegin + leftLen);
//右序数组传入中序右数组 [15,20,7] rootindex + 1=2,inend=5 后续右数组[15,7,20]postBegin + leftLen=1 ,postend - 1=4(排除最后一个节点)
treeNode.right = findNode(inorder, rootindex + 1, inend, postorder, postBegin + leftLen, postend - 1);
return treeNode;
}
}