相同元素的间隔之和

// Java Stream solution
// 思路：
// 1. 使用Java Stream对每个数值及其下标进行分组。
// 2. 计算每个数值的下标间隔之和。
// 3. 将结果转换为数组并返回。

import java.util.*;
import java.util.stream.*;

public class StreamSolution {
    public int[] getIntervals(int[] arr) {
        int n = arr.length;
        Map<Integer, List<Integer>> map = IntStream.range(0, n)
            .boxed()
            .collect(Collectors.groupingBy(i -> arr[i]));

        return IntStream.range(0, n)
            .map(i -> map.get(arr[i]).stream().mapToInt(index -> Math.abs(i - index)).sum())
            .toArray();
    }
}