题解 | #牛群的树形结构重建II#

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param preOrder int整型一维数组
     * @param inOrder int整型一维数组
     * @return TreeNode类
     */
    public TreeNode buildTreeII (int[] preOrder, int[] inOrder) {
        // write code here
        return buildTreeHelper(preOrder, inOrder, 0, 0, inOrder.length - 1);
    }

    private static TreeNode buildTreeHelper(int[] preorder, int[] inorder,
                                            int preIndex, int inStart, int inEnd) {
        if (preIndex >= preorder.length || inStart > inEnd) {
            return null;
        }

        TreeNode root = new TreeNode(preorder[preIndex]);
        int rootIndexInorder = findIndex(inorder, inorder.length, preorder[preIndex]);

        root.left = buildTreeHelper(preorder, inorder, preIndex + 1, inStart,
                                    rootIndexInorder - 1);
        root.right = buildTreeHelper(preorder, inorder,
                                     preIndex + rootIndexInorder - inStart + 1, rootIndexInorder + 1, inEnd);

        return root;
    }

    private static int findIndex(int[] array, int length, int value) {
        for (int i = 0; i < length; i++) {
            if (array[i] == value) {
                return i;
            }
        }
        return -1;
    }

    public static void inorderTraversal(TreeNode root) {
        if (root == null) {
            return;
        }

        inorderTraversal(root.left);
        System.out.print(root.val + " ");
        inorderTraversal(root.right);
    }
}

知识点：

1. 基本的Java语法和概念。
2. 二叉树的构建和遍历。
3. 递归算法的应用。

解题思路：

在这个实现中，我们首先根据先序遍历数组确定根节点，然后在中序遍历数组中找到根节点的位置，从而将数组分为左子树和右子树部分。然后，我们递归地构建左子树和右子树。

buildTreeHelper方法中，preIndex表示当前在先序遍历数组中的索引，inStart和inEnd表示当前中序遍历数组中的起始和结束索引。我们在中序遍历数组中找到根节点的索引，然后根据该索引将数组分为左右子树，并继续递归地构建这些子树。

最终，我们可以通过调用inorderTraversal方法来验证构建的二叉树是否正确。