集美大学-计算机打卡(m)

public TreeNode traversal(List<Integer> inorder, List<Integer> postorder) {
    // 第一步
    if (postorder.isEmpty()) return null;

    // 第二步：后序遍历数组最后一个元素，就是当前的中间节点
    int rootValue = postorder.get(postorder.size() - 1);
    TreeNode root = new TreeNode(rootValue);

    // 叶子节点
    if (postorder.size() == 1) return root;

    // 第三步：找切割点
    int delimiterIndex;
    for (delimiterIndex = 0; delimiterIndex < inorder.size(); delimiterIndex++) {
        if (inorder.get(delimiterIndex) == rootValue) break;
    }

    // 第四步：切割中序数组，得到中序左数组和中序右数组
    List<Integer> inorderLeft = inorder.subList(0, delimiterIndex);
    List<Integer> inorderRight = inorder.subList(delimiterIndex + 1, inorder.size());

    // 第五步：切割后序数组，得到后序左数组和后序右数组
    List<Integer> postorderLeft = postorder.subList(0, delimiterIndex);
    List<Integer> postorderRight = postorder.subList(delimiterIndex, postorder.size() - 1);

    // 第六步
    root.left = traversal(inorderLeft, postorderLeft);
    root.right = traversal(inorderRight, postorderRight);

    return root;
}

//完整代码：优化版本

class Solution {
    Map<Integer, Integer> map;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        map = new HashMap<>();
        for (int i = 0; i < inorder.length; i++) { // 用map保存中序序列的数值对应位置
            map.put(inorder[i], i);
        }

        return findNode(preorder, 0, preorder.length, inorder,  0, inorder.length);  // 前闭后开
    }

    public TreeNode findNode(int[] preorder, int preBegin, int preEnd, int[] inorder, int inBegin, int inEnd) {
        // 参数里的范围都是前闭后开
        if (preBegin >= preEnd || inBegin >= inEnd) {  // 不满足左闭右开，说明没有元素，返回空树
            return null;
        }
        int rootIndex = map.get(preorder[preBegin]);  // 找到前序遍历的第一个元素在中序遍历中的位置
        TreeNode root = new TreeNode(inorder[rootIndex]);  // 构造结点
        int lenOfLeft = rootIndex - inBegin;  // 保存中序左子树个数，用来确定前序数列的个数
        root.left = findNode(preorder, preBegin + 1, preBegin + lenOfLeft + 1,
                            inorder, inBegin, rootIndex);
        root.right = findNode(preorder, preBegin + lenOfLeft + 1, preEnd,
                            inorder, rootIndex + 1, inEnd);

        return root;
    }
}

这段代码实现了根据中序遍历和后序遍历序列构建二叉树的逻辑。下面是对每个步骤的解释：

从中序和后序遍历序列构造二叉树

1. 第一步：如果数组大小为零的话，说明是空节点了。

   • 如果后序遍历数组为空，表示当前节点为 null，直接返回 null。

2. 第二步：如果不为空，那么取后序数组最后一个元素作为节点元素。

   • 如果后序遍历数组不为空，取最后一个元素作为当前节点的值。

3. 第三步：找到后序数组最后一个元素在中序数组的位置，作为切割点。

   • 在中序遍历数组中找到后序数组最后一个元素的位置，确定了当前节点在中序遍历中的位置。

4. 第四步：切割中序数组，切成中序左数组和中序右数组（顺序别搞反了，一定是先切中序数组）。

   • 根据切割点，将中序数组分成左半部分和右半部分，分别表示左子树和右子树。

5. 第五步：切割后序数组，切成后序左数组和后序右数组。

   • 根据切割点，将后序数组分成左半部分和右半部分，分别表示左子树和右子树。

6. 第六步：递归处理左区间和右区间。

   • 递归调用 traversal 方法，分别传入中序左数组和后序左数组构建左子树，以及中序右数组和后序右数组构建右子树。最后将左右子树连接到当前节点。

模板代码

这段代码提供了一个通用的递归模板 traversal 方法，用于根据中序和后序遍历序列构建二叉树。在递归的过程中，根据中序遍历数组的切割点确定当前节点的位置，然后递归处理左右区间。

完整代码：优化版本

这是一个使用了优化的版本，使用了 HashMap 来存储中序遍历数组元素值对应的位置，以提高查找速度。findNode 方法根据前序遍历数组和中序遍历数组的范围递归构建二叉树节点。