Nowcoder java-重建二叉树

Nowcoder java-重建二叉树

（题目来自于牛客网）

递归思路

先序遍历：中左右
中序遍历：左中右
先序遍历的第一个数是根节点，再到中序遍历中根据根节点可以分出来左右子树。在这里确定左子树的个数i-startIn。
左子树的先序遍历起始位置是：startPre +1
左子树的先序遍历结束位置是：startPre + （i - startIn）
左子树的中序遍历起始位置是：startIn
左子树的中序遍历结束位置是：i- 1

右子树的先序遍历起始位置是： startPre + i - startIn +1

右子树的先序遍历结束位置是： endPre
右子树的中序遍历起始位置是： i + 1
右子树的中序遍历结束位置是：endIn

代码

public class ReConstructBinaryTree {
    public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
        TreeNode root = reConstructBinaryTree(pre, 0, pre.length - 1, in, 0, in.length - 1);
        return root;
    }

    public TreeNode reConstructBinaryTree(int[] pre, int startPre, int endPre, int[] in, int startIn, int endIn) {
        if (startPre > endPre || startIn > endIn) {
            return null;
        }
        TreeNode root = new TreeNode(pre[startPre]);
        for (int i = 0; i <= endIn; i++) {
            if (in[i] == pre[startPre]) {
                root.left = reConstructBinaryTree(pre, startPre + 1, startPre + i - startIn, in, startIn, i - 1);
                root.right = reConstructBinaryTree(pre, startPre + i - startIn + 1, endPre, in, i + 1, endIn);
            }
        }
        return root;
    }
}

下图截图上有参数含义，方便理解。