第20天|106.从中序与后序遍历序列构造二叉树 ,105.从前序与中序遍历序列构造二叉树

106.从中序与后序遍历序列构造二叉树

https://leetcode.cn/problems/construct-binary-tree-from-inorder-and-postorder-traversal/submissions/507824940/

/**
 * 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 buildTree(int[] inorder, int[] postorder) {
        int n = postorder.length;
        if (n == 0) { // 空节点
            return null;
        }

        int leftSize = indexOf(inorder, postorder[n - 1]); // 左子树的大小
        int[] in1 = Arrays.copyOfRange(inorder, 0, leftSize);
        int[] in2 = Arrays.copyOfRange(inorder, leftSize + 1, n);
        int[] post1 = Arrays.copyOfRange(postorder, 0, leftSize);
        int[] post2 = Arrays.copyOfRange(postorder, leftSize, n - 1);
        TreeNode left = buildTree(in1, post1);
        TreeNode right = buildTree(in2, post2);
        return new TreeNode(postorder[n - 1], left, right);
    }

    // 返回 x 在 a 中的下标，保证 x 一定在 a 中
    private int indexOf(int[] a, int x) {
        for (int i = 0;; i++) {
            if (a[i] == x) {
                return i;
            }
        }
    }
}

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

. - 力扣（LeetCode）

• 首先，代码获取preorder数组的长度n，如果n为0，则表示当前节点为空节点，直接返回null。
• 接下来，代码通过调用indexOf函数找到preorder在inorder数组中的下标leftSize，即左子树的大小。
• 然后，代码使用Arrays.copyOfRange函数将preorder和inorder数组分成四个部分：pre1、pre2、in1、in2。其中pre1是preorder数组中除去根节点后左子树的部分，pre2是preorder数组中除去根节点后右子树的部分；in1是inorder数组中左子树的部分，in2是inorder数组中右子树的部分。
• 接着，代码递归调用buildTree函数，传入pre1和in1作为参数，构建左子树，并将返回的左子树赋值给变量left。然后，代码再次递归调用buildTree函数，传入pre2和in2作为参数，构建右子树，并将返回的右子树赋值给变量right。
• 最后，代码使用preorder作为值，left作为左子树，right作为右子树，创建一个新的TreeNode对象，并将其作为函数的返回值。

/**
 * 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 buildTree(int[] preorder, int[] inorder) {
         int n = preorder.length;
        if (n == 0) { // 空节点
            return null;
        }

        int leftSize = indexOf(inorder,preorder[0]);
        int[] pre1 = Arrays.copyOfRange(preorder, 1, 1 + leftSize);
                int[] pre2 = Arrays.copyOfRange(preorder, 1 + leftSize, n);
        int[] in1 = Arrays.copyOfRange(inorder, 0, leftSize);
        int[] in2 = Arrays.copyOfRange(inorder, 1 + leftSize, n);
        TreeNode left = buildTree(pre1, in1);
        TreeNode right = buildTree(pre2, in2);
        return new TreeNode(preorder[0], left, right);

    }
       // 返回 x 在 a 中的下标，保证 x 一定在 a 中
    private int indexOf(int[] a, int x) {
        for (int i = 0; ; i++) {
            if (a[i] == x) {
                return i;
            }
        }
    }
}