代码如下，开袋即食
class Solution {
private Map<Integer,Integer> map;
public TreeNode buildTree(int[] preorder, int[] inorder) {
map = new HashMap<>();
for(int i =0;i<preorder.length;i++){
map.put(inorder[i],i);
}
return build(preorder,inorder,0,preorder.length-1,0,preorder.length-1);
}
public TreeNode build(int[] preorder,int[] inorder,int p_left,int p_right,int i_left,int i_right){
if(p_left>p_right||i_left>i_right) return null;
int p_root = p_left;//前序比那里的第一个节点就是根节点
int i_root = map.get(preorder[p_root]);//在中序遍历中定位根节点的位置
TreeNode root = new TreeNode(preorder[p_left]);
int left_size_tree = i_root-p_left;//获得左子树的长度
root.left = build(preorder,inorder,p_left+1,p_left+left_size_tree,i_left,i_root-1);
root.right = build(preorder,inorder,p_left+left_size_tree+1,p_right,i_root+1,i_right);
return root;
}
}
同样这里需要注意前序遍历和中序遍历的左右指针的一个边界问题。
左指针遍历的时候
前序左边界：p_left+1即可
前序右边界：p_left+left_size_tree
中序左边界：i_left
中序右边界：i_root-1
右指针遍历的时候
前序左边界：p_left+left_size_tree+1即可
前序右边界：p_right
中序左边界：i_root+1
中序右边界：i_right
学生所做，记录学习。
另外有一题类似，解法和本题有异曲同工之处。