重建二叉树

Posted 2021-03-11 vvii

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

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

import java.util.Arrays;

public class Solution {
    public TreeNode reConstructBinaryTree(int[] pre, int[] in) {

        if (pre.length == 0) {
            return null;
        }

        // 根
        int root = pre[0];

        //根节点
        TreeNode tree = new TreeNode(root);


        //中序中root值为分割点 左侧：左子树，右侧：右子树
        int index = 0;
        for (; index < in.length - 1; index++) {
            if (in[index] == root) {//在中序遍历中找到根节点
                break;
            }
        }

        //中序遍历中：左子树的起始位置 0
        // 中序遍历中：左子树的结束位置 index
        // copyOfRange(int[] original, int from, int to)  包括from 不包括to

        //前序中pre的首元素为根节点，
        // 第二个元素是左子树的起始位置，
        // 由于左子树个数与中序的个数是一致的，通过左子树个数相等即可推出，前序中左子树的结束位置
        tree.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, index + 1), Arrays.copyOfRange(in, 0, index));

        //中序遍历中：右子树的起始位置 index+1
        // 中序遍历中：右子树的结束位置 in.length-1
        tree.right = reConstructBinaryTree(Arrays.copyOfRange(pre, index + 1, pre.length), Arrays.copyOfRange(in, index + 1, in.length));

        return tree;
    }
}