NC12 重建二叉树

题目描述

给定二叉树的前序遍历和中序遍历，给出原二叉树。如：
输入：{1,2,4,7,3,5,6,8}，{4,7,2,1,5,3,8,6}
输出：{1,2,3,4,#,5,6,#,7,#,#,8}

数据范围在int范围内，且数组内数值不重复。
题目链接：重建二叉树

思路

• 给定二叉树的中序遍历和前序遍历或后续遍历之一可以确定唯一二叉树结构
• 二叉树经典递归
• 递归每一步：找到根结点，左孩子，右孩子
• 出口：当前中序数组或前序数组长度是0

递归

这个过程就是处理两个数组，从两个数组中找到根结点、左子树和右子树。

• 找到根结点，前序数组的第一个元素就是当前根结点的值
• 根据根结点的值r，找到中序数组中r的索引i
• i以左的元素就是该根结点的左子树，以右就是根结点的右子树
• 根据i值划分前序数组和中序数组的子树组，递归左子树和右子树
• 以示例{1,2,4,7,3,5,6,8}，{4,7,2,1,5,3,8,6}为例，第一次递归，1是根结点值，中序数组中1以左使4、7、2，对应前序找到前序数组中2、4、7，不难看出，根据根节点索引i，可以划分左子树为（左闭右开）pre[1,i+1],vin[0,i]，右子树为pre[i+1,len],vin[i+1,len]，以此完成递归建树

public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
        if(pre.length==0 || vin.length==0)return null;
        TreeNode root=new TreeNode(pre[0]);
        for(int i=0;i<vin.length;i++){
            if(pre[0]==vin[i]){
                root.left=reConstructBinaryTree(
                    Arrays.copyOfRange(pre,1,i+1),
                    Arrays.copyOfRange(vin,0,i));
                root.right=reConstructBinaryTree(
                    Arrays.copyOfRange(pre,i+1,pre.length),
                    Arrays.copyOfRange(vin,i+1,vin.length));
                break;
            }
        }
        return root;
    }