剑指offer

涉及的知识点

二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。
输入：

[1,2,3,4,5,6,7],[3,2,4,1,6,5,7]
输出：
{1,2,5,3,4,6,7}

解题思路

二叉树的前序遍历中序遍历都是说的是根节点的位置，所以在前序遍历中的第一个就是根节点，然后根节点将中序遍历分割成左右两个部分，左边就是左子树，右边是右子树，然后再递归建立左右子树。

代码

import java.util.Arrays;
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        //递归结束条件
        if(pre.length==0||in.length==0){
            return null;
        }
        TreeNode root = new TreeNode(pre[0]);
        for(int i=0;i<in.length;i++){
            if(in[i]==pre[0]){
                //左子树，左闭右开
             root.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
                root.right=reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
                break;
            }
        }
        return root;
    }
}