查找大小为 M 的最新分组

/*
 * 使用Java Stream的解法
 * 1. 使用IntStream遍历 arr 数组。
 * 2. 对于每个 arr[i]，更新 steps 数组和哈希表。
 * 3. 每次更新后，检查是否有长度为 m 的区间。
 * 4. 如果存在，记录当前步骤。如果没有，则返回 -1。
 */

import java.util.stream.IntStream;
import java.util.HashMap;

public int findLatestStep(int[] arr, int m) {
    if (arr.length == m) return m;
    int n = arr.length;
    int[] length = new int[n + 2];
    HashMap<Integer, Integer> left = new HashMap<>();
    HashMap<Integer, Integer> right = new HashMap<>();

    int[] result = {-1};

    IntStream.range(0, n).forEach(i -> {
        int pos = arr[i];
        int leftLen = length[pos - 1];
        int rightLen = length[pos + 1];
        int currLen = leftLen + rightLen + 1;
        length[pos] = currLen;
        length[pos - leftLen] = currLen;
        length[pos + rightLen] = currLen;

        left.put(pos - leftLen, pos + rightLen);
        right.put(pos + rightLen, pos - leftLen);

        if (leftLen == m || rightLen == m) {
            result[0] = i;
        }
    });

    return result[0];
}