如何使用Java 8 Stream API改写这段单词统计代码?
使用Java 8 Stream API改写单词统计代码
当然可以用Java 8的Stream API完美替代这段代码,而且写法会更简洁、更符合声明式编程的风格,同时完全保留原代码的功能。
先回顾下原代码的核心逻辑:
将输入字符串转小写,按标点(
?、.)和空白字符分割成单词列表;遍历列表,只统计长度大于2的单词的出现次数,最终返回一个HashMap<String, Integer>类型的统计结果。
下面是两种常用的Stream改写实现:
方式一:使用Collectors.toMap
这种方式直接将单词映射为键,初始计数为1,遇到重复键时累加计数,同时指定用HashMap作为结果容器(和原代码保持一致):
import java.util.function.Function; import java.util.stream.Collectors; public Map<String, Integer> process() { return Arrays.stream(message.toLowerCase().split("[?.\\s]+")) .filter(word -> word.length() > 2) // 过滤长度≤2的单词 .collect(Collectors.toMap( Function.identity(), // 用单词本身作为键 word -> 1, // 每个单词初始计数为1 Integer::sum, // 键重复时累加计数 HashMap::new // 指定结果容器为HashMap )); }
方式二:使用Collectors.groupingBy
如果更偏向“分组统计”的语义,也可以用groupingBy来实现,配合summingInt完成计数:
import java.util.function.Function; import java.util.stream.Collectors; public Map<String, Integer> process() { return Arrays.stream(message.toLowerCase().split("[?.\\s]+")) .filter(word -> word.length() > 2) .collect(Collectors.groupingBy( Function.identity(), // 按单词分组 HashMap::new, // 指定结果容器 Collectors.summingInt(e -> 1) // 统计每组的元素个数 )); }
关键细节说明
- 两种实现都严格保留了原代码的逻辑:转小写、相同的分割正则、过滤长度>2的单词、返回
HashMap - Stream API的写法把**“做什么”和“怎么做”**分离开,不用手动管理循环、Map的
getOrDefault和put操作,代码可读性和可维护性更高 - 如果不需要强制返回
HashMap,可以省略HashMap::new参数,此时会返回默认的Map实现(通常是LinkedHashMap)
内容的提问来源于stack exchange,提问作者Ashlin Karkada
相关产品推荐
相关产品推荐

