剑指 Offer 07. 重建二叉树
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
二叉树前序遍历思想:根节点、左节点、右节点
二叉树中序遍历思想:左节点、根节点、右节点
######我的解题思想:
前序遍历中,第一个节点一定是根节点。到中序遍历中查找这个根节点,根节点之前的为左
树,根节点之后的为右树。(假设输入的前序遍历和中序遍历的结果中都不含重复的数字。)
左右树分开又成了新的两颗二叉树。这里根节点是3,在中序中3前面是9,9为左树,后
面是20、15、7,此为右树。以此递归下去,直到拆分的左右两边都为0为终止条件。
看代码:
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
//把前序遍历的值和中序遍历的值放到list中
List<Integer> preorderList = new ArrayList<>();
List<Integer> inorderList = new ArrayList<>();
for (int i = 0; i < preorder.length; i++) {
preorderList.add(preorder[i]);
inorderList.add(inorder[i]);
}
return helper(preorderList, inorderList);
}
private TreeNode helper(List<Integer> preorderList, List<Integer> inorderList) {
if (inorderList.size() == 0)
return null;
//前序遍历的第一个值就是根节点
int rootVal = preorderList.remove(0);
//创建跟结点
TreeNode root = new TreeNode(rootVal);
//查看根节点在中序遍历中的位置,然后再把中序遍历的数组劈两半,前面部分是
//根节点左子树的所有值,后面部分是根节点右子树的所有值
int mid = inorderList.indexOf(rootVal);
//[0,mid)是左子树的所有值,inorderList.subList(0, mid)表示截取inorderList
//的值,截取的范围是[0,mid),包含0不包含mid。
root.left = helper(preorderList, inorderList.subList(0, mid));
//[mid+1,inorderList.size())是右子树的所有值,
// inorderList.subList(mid + 1, inorderList.size())表示截取inorderList
//的值,截取的范围是[mid+1,inorderList.size()),包含mid+1不包含inorderList.size()。
root.right = helper(preorderList, inorderList.subList(mid + 1, inorderList.size()));
return root;
}
}