题目:leetcode106. 从中序与后序遍历序列构造二叉树
描述:
给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。
思路
这道题使用中序和后序遍历构造二叉树,我们可以根据后序遍历顺序的最后一个元素确定根节点,然后根据这个根节点把中序遍历序列分为两个子序列和一个根节点root,这两个子序列分别是根节点的左子树和右子树。所以我们就完成了第一个结点的配置。然后左子树和右子树又分别重复刚刚的事情。
代码
import java.util.HashMap;
import java.util.Map;
class Solution {
Map<Integer,Integer> map;
public TreeNode buildTree(int[] inorder, int[] postorder) {
map=new HashMap<>();
for (int i = 0; i < inorder.length; i++) {
map.put(inorder[i],i);
}
return createTree(inorder,0,inorder.length,postorder,0,postorder.length);
}
private TreeNode createTree(int[] inorder,int ibegin,int iend,int[] postorder,int postbegin,int postend)
{
if(ibegin>=iend || postbegin>=postend)
return null;
int rootIndex=map.get(postorder[postend-1]);
TreeNode root=new TreeNode(inorder[rootIndex]);
int lenLeft=rootIndex-ibegin;
root.left=createTree(inorder,ibegin,rootIndex,postorder,postbegin,postbegin+lenLeft);
root.right=createTree(inorder,rootIndex+1,postend-1,postorder,postbegin+lenLeft,postend-1);
return root;
}
}