在Java8中使用嵌套for循环，找出给定的差异[duplicate]

Question

我有下面的代码，我使用嵌套的循环，我有一些条件，打破了内部的循环，这提高了这段代码的性能。

假设提供的列表已排序。现在我想找到元素的数量，它们的差值等于某个值，比如说k。

public static int getCount(List<Integer> list, int k) {
    int result = 0;
    for (int i = 0; i < list.size(); i++) {
        for (int j = i + 1; j < list.size(); j++) {
            if (list.get(j) - list.get(i) > k) {
                break;
            }
            if (list.get(j) - list.get(i) == k) {
                result++;
            }
        }
    }
    return result;
}

现在使用相同的逻辑Java8个流，这里为了跳过内部循环我已经使用了返回语句，但是由于内部循环没有被破坏，性能没有提高。

public static int getCount(List<Integer> list, int k) {
    int[] result = { 0 };
    IntStream.range(0, list.size()).forEach(i -> {
        IntStream.range(i + 1, list.size()).forEach(j -> {
            if (list.get(j) - list.get(i) >= k)
                return;
            if (list.get(j) - list.get(i) == k) {
                result[0]++;
            }
        });
    });
    return result[0];
}

Answer 1

用以下内容替换您的内部IntStream.for每个（），看看它是否如预期的那样工作：

IntStream.range(i + 1, list.size())
        .filter(j -> list.get(j) - list.get(i) == k)
        .forEach(j -> result[0]++);