删除值以只保留N个事件

Question

{1,2,3,4,1}, N=1 -> {1,2,3,4}
{2,2,2,2}, N=2 -> {2,2}

 public static int[] deleteNth(int[] elements, int maxOcurrences) {
    List<Integer> ints = Arrays.stream(elements)
                               .boxed()
                               .collect(Collectors.toList());
    return ints.stream().filter(x -> Collections.frequency(ints, x) <= maxOcurrences)
        .mapToInt(Integer::intValue)
        .toArray();
}

PS：我知道O(n^2)，但这对我来说不是问题。

Answer 1

我认为这是一个不使用流的好例子。当涉及有状态操作时，流并不总是最好的方法。

但这是绝对可以做到的，而且这个问题还专门针对流提出，所以您可以使用以下内容。

使用forEachOrdered

public static int[] deleteNth(int[] elements, int maxOcurrs) {
    List<Integer> list = new ArrayList<>();
    Arrays.stream(elements).forEachOrdered(elem -> {
        if (Collections.frequency(list, elem) < maxOcurrs) list.add(elem);
    });
    return list.stream().mapToInt(Integer::intValue).toArray();
}

当流是有序的和连续的时，例如arrays.stream（元素）.boxed()，collect()方法不使用组合器操作符（对于java8和java9当前的realease，这是事实，但是不能保证在下一个版本中完全相同，因为可能会进行许多优化）。

这个实现保持了流的顺序，正如前面提到的，在当前的版本中可以很好地工作。就像下面链接中的答案所说的那样，我个人也认为，在顺序流中实现collect时需要使用组合器是非常困难的。

collect方法的代码如下：

public static int[] deleteNth(int[] elements, int maxOcurrs) {
    return Arrays.stream(elements).boxed()
            .collect(() -> new ArrayList<Integer>(),
                    (list, elem) -> {
                        if (Collections.frequency(list, elem) < maxOcurrs) list.add(elem);
                    },
                    (list1, list2) -> { 
                        throw new UnsupportedOperationException("Undefined combiner"); 
                    })
            .stream()
            .mapToInt(Integer::intValue)
            .toArray();
}

这个收集器创建一个ArrayList，并在添加新元素时检查是否满足MaxOcurrences，如果不满足，则添加该元素。正如前面提到的，在下面的回答中，合并器根本没有被调用。这比n^2好一点。

关于为什么在序列流中不调用组合器方法的更多信息可以在这里找到。