从前序与中序遍历序列构造二叉树

代码如下，开袋即食

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

学生所做，记录学习。