根据前中序构建树的代码实现

1.理论部分知识

树的前序中序后序遍历

---

2.前中序构建树的代码实现

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) {
        val = x;
    }
}

public class Pre_In {
    public TreeNode reConstructBinaryTree(int[] pre, int[] in){//pre是前序遍历数组，in是后序遍历数组
        int len = pre.length;
        TreeNode root = new TreeNode(pre[0]);//根据前序遍历顺序，pre[0]为根节点
        if(pre.length == 1){
            root.left = null;
            root.right = null;
            return root;
        }

        int flag = 0;
        for(int i = 0; i < len; i++){
            if(pre[0] == in[i]){
                flag = i;
                break;
            }
        }

        //flag == 0说明没有左子树
        if(flag > 0){
            //左子树的前序数组
            int[] leftPre = new int[flag];
            //左子树的中序数组
            int[] leftIn = new int[flag];
            for(int j = 0; j < flag; j++){
                leftPre[j] = pre[j + 1];   //前序遍历第一个节点是根节点
            }
            for(int j = 0; j < flag; j++){
                leftIn[j] = in[j];
            }
            root.left = reConstructBinaryTree(leftPre,leftIn);

        } else{
            root.left = null;
        }


        if(pre.length - flag - 1 > 0){
            int[] rightPre = new int[pre.length - 1 - flag];  //右子树长度为总长度减去左子树长度和根节点长度为1
            int[] rightIn = new int[pre.length - 1 - flag];
            for(int j = flag + 1; j < len; j++){  //前面有1一个根节点和flag个左子树，所以从flag + 1开始
                rightIn[j - flag -1] = in[j];   //j - flag -1这样减出来就是从0开始
                rightPre[j - flag - 1] = pre[j];
            }
            root.right = reConstructBinaryTree(rightPre,rightIn);
        } else{
            root.right = null;
        }
        return root;
    }

}