106. 从中序与后序遍历序列构造二叉树:
题目链接 :106. 从中序与后序遍历序列构造二叉树
题目:
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
思路:
- 本体和105为姊妹题,递归方法来说目的一致:找到根节点和其左右子树在两种序列中的范围。
- 首先根据后序最后一个节点找到root,然后根据中序遍历的规律,root值对应Index的左侧为其左子树,右侧为其右子树,那么可以判断左右子树的范围。
- 根据中序得到的左子树序列去寻找后序中的左右子树范围,后序遍历是左右根,所以也比较好找到。
AC代码:
class Solution {
HashMap<Integer,Integer> map = new HashMap<>();
int[] post;
public TreeNode buildTree(int[] inorder, int[] postorder) {
for(int i = 0;i < inorder.length; i++) map.put(inorder[i], i);
post = postorder;
TreeNode root = buildTree(0, inorder.length - 1, 0, post.length - 1);
return root;
}
public TreeNode buildTree(int is, int ie, int ps, int pe) {
if(ie < is || pe < ps) return null;
int root = post[pe];
int rootIdx = map.get(root);
TreeNode node = new TreeNode(root);
node.left = buildTree(is, rootIdx - 1, ps, ps + rootIdx - is - 1);
node.right = buildTree(rootIdx + 1, ie, ps + rootIdx - is, pe - 1);
return node;
}
}