使用 Java 集合对对象列表进行分组和计数

Question

哪个Java Collection 类更适合对对象列表进行分组？

我有一个来自用户的消息列表，如下所示：

aaa hi
bbb hello
ccc Gm
aaa  Can?
CCC   yes
ddd   No

从这个消息对象列表中，我想计算并显示aaa(2)+bbb(1)+ccc(2)+ddd(1)。任何代码帮助？

Answer 1

您可以使用Map<String, Integer>，其中键代表单个字符串，映射值是每个字符串的计数器。

所以你可以这样做：

// where ever your input comes from: turn it into lower case,
// so that "ccc" and "CCC" go for the same counter
String item = userinput.toLowerCase(); 

// as you want a sorted list of keys, you should use a TreeMap
Map<String, Integer> stringsWithCount = new TreeMap<>();
for (String item : str) {
  if (stringsWithCount.contains(item)) {
    stringsWithCount.put(item, stringsWithCount.get(item)+1));
  } else {
    stringsWithCount.put(item, 0);
  }
}

然后你可以在完成后迭代地图：

for (Entry<String, Integer> entry : stringsWithCount.entrySet()) {

并构建您的结果字符串。

这就像老式的实现；如果你想花哨并给你的老师一个惊喜，你可以选择 Java8/lambda/stream 解决方案。 （除非您真的花时间完全理解以下解决方案，否则我不建议这样做；因为这在我这边未经测试）

Arrays.stream(someListOrArrayContainingItems)
  .collect(Collectors
     .groupingBy(s -> s, TreeMap::new, Collectors.counting()))
  .entrySet()
  .stream()
  .flatMap(e -> Stream.of(e.getKey(), String.valueOf(e.getValue())))
  .collect(Collectors.joining())

Answer 2

将其他几个答案的部分拼凑在一起，根据其他问题调整您的代码并修复一些小错误：

    // as you want a sorted list of keys, you should use a TreeMap
    Map<String, Integer> stringsWithCount = new TreeMap<>();
    for (Message msg : convinfo.messages) {
        // where ever your input comes from: turn it into lower case,
        // so that "ccc" and "CCC" go for the same counter
        String item = msg.userName.toLowerCase();
        if (stringsWithCount.containsKey(item)) {
            stringsWithCount.put(item, stringsWithCount.get(item) + 1);
        } else {
            stringsWithCount.put(item, 1);
        }
    }
    String result = stringsWithCount
            .entrySet()
            .stream()
            .map(entry -> entry.getKey() + '(' + entry.getValue() + ')')
            .collect(Collectors.joining("+"));
    System.out.println(result);

打印出来：

aaa(2)+bbb(1)+ccc(2)+ddd(1)

Answer 3

你需要一个来自番石榴的MultiSet。该集合类型是为此类任务量身定制的：

MultiSet<String> multiSet = new MultiSet<>();
for (String line : lines) { // somehow you read the lines
    multiSet.add(line.split(" ")[0].toLowerCase());
}
boolean first = true;
for (Multiset.Entry<String> entry : multiset.entrySet()) {
    if (!first) {
        System.out.println("+");
    }
    first = false;
    System.out.print(entry.getElement() + "(" + entry.getCount() + ")");            
}

Answer 4

假设您使用 Java 8，使用 Stream API 可能是这样的：

List<Message> messages = ...;
// Convert your list as a Stream
// Extract only the login from the Message Object
// Lowercase the login to be able to group ccc and CCC together
// Group by login using TreeMap::new as supplier to sort the result alphabetically
// Convert each entry into login(count)
// Join with a +
String result =
    messages.stream()
        .map(Message::getLogin)
        .map(String::toLowerCase)
        .collect(
            Collectors.groupingBy(
                Function.identity(), TreeMap::new, Collectors.counting()
            )
        )
        .entrySet()
        .stream()
        .map(entry -> entry.getKey() + '(' + entry.getValue() + ')')
        .collect(Collectors.joining("+"))
System.out.println(result);

输出：

aaa(2)+bbb(1)+ccc(2)+ddd(1)

如果您想通过登录对消息进行分组并将结果作为一个集合，您可以继续下一步：

Map<String, List<Message>> groupedMessages = 
    messages.stream()
        .collect(
            Collectors.groupingBy(
                message -> message.getLogin().toLowerCase(), 
                TreeMap::new, 
                Collectors.toList()
            )
        );