105. 从前序与中序遍历序列构造二叉树

力扣打卡：105. 从前序与中序遍历序列构造二叉树

具体的看思路

对于一个父节点和两个子节点（一个节点可以是一棵树）

一定要熟记各种前中后遍历的顺序

复习：

• 前序遍历是：从父节点到左节点到右节点，也就是中左右
• 中序遍历是：从左节点到父节点到右节点，也就是左中右
• 后序遍历是：从左节点到右节点到父节点，也就是左右中

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */

class Solution {
    public TreeNode buildTreeB(int[] preorder, int[] inorder) {
        // 通过一个辅助函数来构建这个二叉树，因为要控制数组的索引位置
        return build(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
    }
    // 要控制两个数组的索引，那么最少需要四个参数，两个数组索引位置的开始和结束，此时再加入两个参数，两个数组
    public TreeNode buildB(int[] preorder,int preStart,int preEnd,
                        int[] inorder,int inStart, int inEnd){
        
        if(preStart>preEnd) return null;
        // 因为前序遍历中,第一个就是中,也就是父节点,所以控制好前序遍历数组的第一个元素也就是当前的父节点
        int rootVal = preorder[preStart];
        TreeNode root = new TreeNode(rootVal); // 生成了当前的父节点

        // 因为前序数组中,无法确定左子树和右子树,但是再中序数组中,因为遍历时左中右,如果确定了父节点,那么左右字数都可以确定了
        int index = 0;
        for(int i=inStart; i<=inEnd; i++){  // 根据中序遍历的特点,找到左子树的长度leftSize
            if(rootVal==inorder[i]) {index=i; break;}
        }
        int leftSize = index - inStart;

        // 当前的函数已经成功的定义了,所以递归的调用构建子树
        root.left = build(preorder,preStart+1,preStart+leftSize,inorder,inStart,index-1);
        root.right = build(preorder,preStart+leftSize+1,preEnd,inorder,index+1,inEnd);

        return root;
    }

}

// 重新复习一遍
 public TreeNode buildTree(int[] preorder, int[] inorder){
     // 前序遍历的顺序是：在当前节点中：父节点->左节点->右节点，其中左节点为左子树，右节点为右子树 即整体顺序是中左右，但是需要递归到最后一个
     // 中序遍历的顺序是：在当前节点中，左节点-> 父节点 -> 右节点 可以画图和伪代码的形式进行观察
     // 后续遍历的顺序是：在当前节点中，左节点 -> 右节点 -> 父节点 
     // 因为操作是在子节点和右节点后的，结合basecase，应该是做节点先记录，然后是右节点，最后返回上一层的父节点
     
     // 看各种层级的遍历顺序，需要看具体的操作在哪

     // 综上而言：因为前序遍历的第一个就是父节点，而中序遍历的中间某一个就是父节点，左边的是左子树，右边的是右子树，所以根据这个进行构建
     return build(preorder,0,preorder.length-1,inorder,0,inorder.length-1);

 }
 public TreeNode build(int[] preorder, int preStart, int preEnd,
                     int[] inorder, int inStart, int inEnd){

     // 最后确定终止条件
     if(preStart>preEnd) return null;

     // 首先确定参数，有两个数组，preorder，inorder，同时为了寻找每一个父节点，左子树，右子树，那么两个数组都需要开始索引和截至索引
     // 所以总共需要六个参数

     // 开始写逻辑,basecase应该在逻辑确定后进行确定
     // 每一次前序遍历的第一个节点就是父节点,那么将父节点构建出来,然后构建左右子树
     int rootVal = preorder[preStart];
     TreeNode root = new TreeNode(rootVal);

     // 因为中序遍历中,左边是左子树,中间某一个是父节点,右边是左右子树,所以找到父节点,就能得到左子树,右子树
     // 也即:在一个给定的索引范围内找到父节点,然后分割左子树和右子树,继续递归构建
     int point = 0;
     for(int i=inStart; i<=inEnd; i++){
         if(inorder[i]==rootVal) { point = i; break;}
     }

     // 已经找到了父节点的位置,那么开始递归的调用方法,分别构建子树,对于不同的节点,索引范围不一样
     // 对于前序遍历数组来说,因为父节点就是当前了,所以下一个树的父节点就是prestart+1
     // 同时截至位置是通过中序数组来计算的,因为前序遍历的顺序是:中左右
     // 而中序遍历的顺序是左中右(先递归到最左边,记录左节点,然后返回上一层,记录父节点,最后进入下一层记录右节点)
     // 所以可以通过两个数组来确定左子树的长度

     // 也就是:
     int leftSize = point - inStart; // 此时即为左子树的长度


     // 前序数组的截至索引就为左子树这个范围,也就是preStart+1,preStart+leftSize
     // 中序数组的索引范围为:从instart开始,到父节点位置,就是左子树的范围,进行递归的构建
     root.left = build(preorder,preStart+1,preStart+leftSize,
                     inorder,inStart,point-1);
     root.right = build(preorder,preStart+leftSize+1,preEnd,
                     inorder,point+1,inEnd);

     // 最后考虑basecase
     // 因为对于每一个前序数组的元组而言，每个元素都可以看成是一个父节点，所以只要左子树的索引范围超过了右子树，那么就表明了，终止
     // 写在第一个行位置
     return root;

 }