剑指 Offer - 4：重建二叉树

题目描述

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

题目链接：https://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6

解题思路

根据前序遍历的特点，我们可以通过第一个值得到当前子树的根节点，借此去中序遍历查找根节点的左子树和右子树，采用递归的思路

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        // 获取根节点
        int val = pre[0];
        TreeNode root = new TreeNode(val);
        // 得到左子树的大小
        int len = 0;
        for (; len < in.length; len++) {
            if (in[len] == val) break;
        }
        // 构建左子树
        if (len != 0) {
            // 左子树的先序遍历
            int[] pre_left = new int[len];
            for (int i = 0; i < len; i++) {
                pre_left[i] = pre[i+1];
            }
            // 左子树的中序遍历
            int[] in_left = new int[len];
            for (int i = 0; i < len; i++) {
                in_left[i] = in[i];
            }
            root.left = reConstructBinaryTree(pre_left, in_left);
        }
        
        // 构建右子树
        if (len != (pre.length-1)) {
            // 右子树的先序遍历
            int[] pre_right = new int[pre.length-len-1];
            for (int i = 0, idx = len+1; idx < pre.length; i++, idx++) {
                pre_right[i] = pre[idx];
            }
            // 右子树的中序遍历
            int[] in_right = new int[in.length-len-1];
            for (int i = 0, idx = len+1; idx < in.length; i++, idx++) {
                in_right[i] = in[idx];
            }
            root.right = reConstructBinaryTree(pre_right, in_right);
        }
        return root;
    }
}