思路:对于preorder = [3,9,20,15,7]、 inorder = [9,3,15,20,7]这两个数组,易知前序遍历的首个数组元素为中序遍历的根节点,因此3为根节点,而3左边的元素为左子树元素,右边的元素为右子树元素,并且对于inorder 数组中3的左边1个元素为左子树的中序遍历元素集合,右边3个元素为右子树的中序遍历元素集合,对于preorder 数组中3的左边1个元素9为左子树的前序遍历元素集合,9右边的3个元素为右子树的前序遍历元素集合。因此可以使用递归来解题,将问题规模不断拆分。代码如下:
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder.length == 0 || inorder.length == 0){
return null;
}
TreeNode treeNode = new TreeNode(preorder[0]);
for(int i = 0; i < inorder.length; ++i){
if(preorder[0] == inorder[i]){
treeNode.left = buildTree(Arrays.copyOfRange(preorder, 1, i + 1),
Arrays.copyOfRange(inorder, 0, i));
treeNode.right = buildTree(Arrays.copyOfRange(preorder, i + 1, preorder.length),
Arrays.copyOfRange(inorder, i + 1, inorder.length));
break;
}
}
return treeNode;
}
}