重建二叉树

题目描述

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

 

 

根据前序和中序确定二叉树，根据它们的特性，可知前序第一个元素是根节点，和中序对应找出各自的左子树和右子树。

然后递归循环，在找相同的元素，此为子树的根节点，然后继续操作，直到子树为空。

 

  Arrays.copyOfRange(byte[] original, int from, int to)是对数组的指定位置的复制。

 

下面是实现的代码：

 

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.*;

public class Solution {
    public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
        if (pre == null || in == null || pre.length == 0 || in.length == 0) {
            return null;
        }
        TreeNode root = construct(pre, in);
        return root;
    }

    private TreeNode construct(int[] pre, int[] in) {
        int length = pre.length;
        if (length == 0)
            return null;
        TreeNode root = new TreeNode(pre[0]);
        int i = 0;
        for (; i < length; i++) {
            if (in[i] == pre[0])
                break;
        }
        if (i == length)
            i--;
        int[] leftPre = Arrays.copyOfRange(pre, 1, i + 1);
        int[] rightPre = Arrays.copyOfRange(pre, i + 1, length);
        int[] leftIn = Arrays.copyOfRange(in, 0, i);
        int[] rightIn = Arrays.copyOfRange(in, i + 1, length);
        root.left = construct(leftPre, leftIn);
        root.right = construct(rightPre, rightIn);
        return root;
    }
}