Java计数整数数组中每个元素的出现次数

时间:2016-08-14 12:02:47

标签: java java-8 java-stream

我编写了以下代码段来计算每个元素的出现次数。是否有可能以更短的方式实现这一目标?

int[] arr = {1, 6, 2, 8, 5, 4, 7, 7, 5, 7};
Arrays.stream(arr)
        .collect(ArrayList::new, ArrayList::add, ArrayList::addAll)
        .stream()
        .collect(Collectors.groupingBy(s -> s))
        .forEach((k, v) -> System.out.println(k+" "+v.size()));

另外,我想只显示超过1次的元素。所以我尝试修改如下导致错误。

.forEach((k, v) -> if(v.size() > 1) System.out.println(k+" "+v.size()));

这样做的正确方法是什么?

5 个答案:

答案 0 :(得分:8)

对于后一个问题,你必须改变

.forEach((k, v) -> if(v.size() > 1) System.out.println(k+" "+v.size()));

.forEach((k, v) -> {if(v.size() > 1) System.out.println(k+" "+v.size());});

对于第一部分,不清楚为什么需要第一个collect后跟第二个Stream管道。 如果目的是将IntStream转换为Stream<Integer>,请使用boxed()

Arrays.stream(arr)
      .boxed()
      .collect(Collectors.groupingBy(s -> s))
      .forEach((k, v) -> System.out.println(k+" "+v.size()));

正如Dici建议的那样,你也可以将收集者链接起来,将每个数字与其出现次数分组:

Map<Integer,Integer> occurrences = 
    Arrays.stream(arr)
          .boxed()
          .collect(Collectors.groupingBy(s -> s, Collectors.counting()));

答案 1 :(得分:3)

如果您愿意使用第三方库,Eclipse CollectionsBag类型可以按以下方式使用:

Bags.mutable.with(1, 6, 2, 8, 5, 4, 7, 7, 5, 7)
    .selectByOccurrences(count -> count > 1)
    .forEachWithOccurrences((k, count) -> System.out.println(k+" "+count));

如果必须将int [] arr变量保存为int数组,则可以使用IntBag更新:2016年9月9日:我已将selectByOccurrences添加到原始Bags,并在EC 8.0 release中提供。以下代码现在可以正常工作,而无需使用collect(i -> i)

打包整数
IntBags.mutable.with(arr)
    .selectByOccurrences(count -> count > 1)
    .forEachWithOccurrences((k, count) -> System.out.println(k+" "+count));

注意:我是Eclipse Collections的提交者。

答案 2 :(得分:0)

我也希望分享我的解决方案!!!

// Solution 1 [Improved from Eran's solution & suggestion]
int[] arr = {1, 6, 2, 8, 5, 4, 7, 7, 5, 7};
Map<Integer, Long> counts = Arrays.stream(arr)
    .boxed()
    .collect(collectingAndThen(groupingBy(n -> n, counting()),
        map -> map.entrySet().stream()
            .filter(n -> n.getValue() > 1)
            .collect(toMap(Entry::getKey, Entry::getValue))
));
System.out.println(counts.toString());

// Solution 2 [Improved from Dici's suggestion]
int[] arr = {1, 6, 2, 8, 5, 4, 7, 7, 5, 7};
Map<Object, Long> counts = Arrays.stream(arr)
    .collect(ArrayList::new, ArrayList::add, ArrayList::addAll)
    .stream()
    .collect(groupingBy(Function.identity(), counting()));
counts.values().removeIf(count -> count < 2);
System.out.println(counts.toString());  

答案 3 :(得分:0)

也可以使用频率来完成:

 List<Integer> list = ImmutableList.of(1, 2, 3, 4, 5, 6, 3, 4, 5);
 Map<Integer, Integer> result = list.stream().distinct().collect(Collectors.toMap(Function.identity(), token -> Collections.frequency(list, token)));

答案 4 :(得分:0)

List<Integer> numsList=Arrays.asList(1,2,3,5,2,4,3,1,2,2,4,4,5);
Map<Integer, Long> map=numsList.stream().collect(Collectors.groupingBy(Integer::intValue,Collectors.counting()));
map.forEach((k,v)->{System.out.println(k+" = "+v);});