根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
思路:
1、通过中序遍历和后续遍历可以看出,后续遍历中最后一个节点3为跟节点。
2、根据这个节点可以将中序遍历数组分为两个左右子树数组 [9] [15,20,7]
3、通过对中序遍历数组划分,可以直到左子树节点个数,然后对后续遍历数组进行划分[9] [15,7,20]
4、后序遍历数组划分完成后,可以分别根据划分后的数组最后一个节点,构建子树中的根节点,9,20。
5、一次递归构建,则可完成
class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
int instart = 0, inend = inorder.length - 1;
int poststart = 0, postend = postorder.length - 1;
if(postorder == null || postend == -1) return null;
return create(inorder,instart,inend,postorder,poststart,postend);
}
public TreeNode create(int [] inorder, int instart, int inend, int[] postorder, int poststart, int postend){
if(postend < poststart || instart > inend) return null;
int value = postorder[postend]; //后续遍历的最后一个节点作为根节点,记录值
int mid = 0;
for(int i = 0; i < inorder.length; i++){ //后序遍历的最后一个节点在中序遍历中的位子。
if(value == inorder[i]){
mid = i;
}
}
int num = mid - instart; //记录左子树节点个数,后面用于构造子树时 数组区间下标设置。
TreeNode root = new TreeNode(value);
root.left = create(inorder,instart,mid - 1,postorder,poststart,poststart + num - 1);
root.right = create(inorder,mid + 1,inend,postorder,poststart + num,postend - 1);
return root;
}
}