剑指 Offer 07. 重建二叉树–中等
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/
9 20
/
15 7
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
这道题先要了解前序遍历以及中序遍历的性质;
即
前序遍历:根节点,左子树,右子树
中序遍历:左子树,根节点,右子树
了解之后 可以通过比较两个数组,确定中序遍历中根节点的位置,然后左边就是左子树,右边就是右子树,再通过左右子树的长度去前序遍历中确定左子树和右子树的位置,递归调用即可实现
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder.length == 0){
return null;
}
// 根节点
TreeNode root = new TreeNode(preorder[0]);
//确定根节点在中序遍历中的位置
int index = 0;
for(int i = 0; i < inorder.length; i++){
if(inorder[i] == preorder[0]){
index = i;
}
}
//通过中序遍历计算出左右子树数组的长度在前序遍历中应用
//左子树
root.left = buildTree(Arrays.copyOfRange(preorder, 1, index + 1), Arrays.copyOfRange(inorder, 0, index));
//右子树
root.right = buildTree(Arrays.copyOfRange(preorder, index + 1, preorder.length), Arrays.copyOfRange(inorder, index+1, inorder.length));
return root;
}
以上~