相信我，使用 Stream 真的可以让代码更优雅！

点击上方“芋道源码”，选择“设为星标”

管她前浪，还是后浪？

能浪的浪，才是好浪！

每天 10:33 更新文章，每天掉亿点点头发...

源码精品专栏

• 原创 | Java 2021 超神之路，很肝~
• 中文详细注释的开源项目
• RPC 框架 Dubbo 源码解析
• 网络应用框架 Netty 源码解析
• 消息中间件 RocketMQ 源码解析
• 数据库中间件 Sharding-JDBC 和 MyCAT 源码解析
• 作业调度中间件 Elastic-Job 源码解析
• 分布式事务中间件 TCC-Transaction 源码解析
• Eureka 和 Hystrix 源码解析
• Java 并发源码

来源：blog.csdn.net/qq_41698074/

article/details/108502976/

• 前言
• 技巧
  • 数组转集合
  • 统计数组元素中的个数
  • 基本数据类型的数组自定义排序
  • 统计数组中前 k 个个高频元素
• 总结

---

前言

虽然 stream在 Java8 中就已经被引入，但是大多数人却没有去使用这个十分有用的特性，本文就通过介绍几个通过使用stream让代码更简洁、可读，来让你了解stream的方便之处。

基于 Spring Boot MyBatis Plus Vue & Element 实现的后台管理系统 用户小程序，支持 RBAC 动态权限、多租户、数据权限、工作流、三方登录、支付、短信、商城等功能

• 项目地址：https://gitee.com/zhijiantianya/ruoyi-vue-pro
• 视频教程：https://doc.iocoder.cn/video/

技巧

数组转集合

相信经常刷LeetCode的小伙伴，偶尔会遇到需要将List与基本类型数组进行互转的情况，然后就需要写像下面这样的代码：

代码语言：javascript复制

// 将 List 元素存储到数组中
List<Integer> list = new ArrayList<>(Arrays.asList(1, 2, 3, 4, 5));
int[] arr = new int[list.size()];
Integer[] temp = list.toArray(new Integer[0]);
for (int i = 0; i < temp.length; i  ) {
 arr[i] = temp[i];
}

// 将数组元素 存储到 List 中
int[] arr = {1, 2, 3, 4, 5};
List<Integer> list = new ArrayList<>();
for (int val : arr) {
 list.add(val);
}

以上两个转换虽然写着还不算麻烦，但是每次都需要写一个循环，尤其在数组转List的时候还需要使用一个临时数组，都会让人看着很不舒服，但是如果使用了stream就会大不一样，用stream实现了相同功能的代码如下：

代码语言：javascript复制

// 将 List 元素存储到数组中
List<Integer> list = new ArrayList<>(Arrays.asList(1, 2, 3, 4, 5));
int[] arr = list.stream().mapToInt(Integer::intValue).toArray();

// 将数组元素 存储到 List 中
int[] arr = {1, 2, 3, 4, 5};
List<Integer> list = IntStream.of(arr).boxed().collect(Collectors.toList());

可以发现通过使用stream，我们能够在写代码的时候更加连贯，代码也更加可靠易维护，注意力也可以放在业务功能上，相信各位就算对lambda语法并不是太熟悉，在阅读上面代码的时候，也很容易能够看懂。

统计数组元素中的个数

假设我们现在需要统计并输出一个有重复元素的数组中每个元素及对应元素出现的个数，相信各位都能够想到，我们使用一个Map就很容易解决这个问题，代码如下：

代码语言：javascript复制

String[] arr = {"a", "c", "a", "b", "d", "c"};
Map<String, Integer> map = new HashMap<>();
for (String s : arr) {
    if (map.containsKey(s)) {
        map.put(s, map.get(s)   1);
    } else {
        map.put(s, 1);
    }
}
map.forEach((key, value) -> System.out.println(key   " : "   value));

如果对Map中的API更加熟悉的小伙伴，可能会写出下面这个更加简洁的代码：

代码语言：javascript复制

String[] arr = {"a", "c", "a", "b", "d", "c"};
Map<String, Integer> map = new HashMap<>();
for (String s : arr) {
    map.put(s, map.getOrDefault(s, 0)   1);
}
map.forEach((key, value) -> System.out.println(key   " : "   value));

但是，如果使用stream，我们还能写出更加简洁的代码，同样不需要写烦人的循环了，而且只需两行代码即可(为了提高可读性，进行了换行)：

代码语言：javascript复制

String[] arr = {"a", "c", "a", "b", "d", "c"};
Stream.of(arr)
      .collect(Collectors.toMap(k -> k, k -> 1, Integer::sum))
      .forEach((k, v) -> System.out.println(k   " : "   v));

注意

在上面的代码中，Collectors.toMap(k -> k, k -> 1, Integer::sum)这一部分可能不好理解，对于这里面的三个参数，第一个参数代表将arr中的每一个元素作为Map中的key，第二个参数代表每一个key所对应的value，在这里每一个元素都对应个数1，第三个参数代表，如果存在相同的key，该如何进行合并，这里通过使用Integer::sum，代表将具有相同key的元素进行合并时，其value进行相加，这样便实现了每个元素个数的统计。

基本数据类型的数组自定义排序

有时我们会遇到对基本数据类型的数组进行自定义排序的情况，不同于包装类型的数组和集合可以直接使用比较器，我们只能通过将基本数组类型的数组转为包装类型或者存储在集合中，在排序完成后再转为基本类型的数组，再者，我们只能通过手写排序算法，修改排序算法中的比较进行实现。

不管是哪种方法，我们都没办法将精力放在逻辑功能上，必须写一些额外的代码，甚至是修改底层逻辑，就像下面的代码一样（实现数组逆序）：

代码语言：javascript复制

int[] arr = {1, 5, 9, 7, 2, 3, 7, -1, 0, 3};
// 将数组转为包装类型再进行自定义排序
Integer[] temp = new Integer[arr.length];
for (int i = 0; i < arr.length; i  ) {
    temp[i] = arr[i];
}
Arrays.sort(temp, Comparator.reverseOrder());
for (int i = 0; i < temp.length; i  ) {
    arr[i] = temp[i];
}

// 将数组转为集合类型再进行自定义排序
List<Integer> list = new ArrayList<>();
for (int val : arr) {
    list.add(val);
}
list.sort(Collections.reverseOrder());
for (int i = 0; i < list.size(); i  ) {
    arr[i] = list.get(i);
}

// 通过手写排序算法修改比较规则实现
// 为了让代码更加简洁，使用了最暴力且没有优化的冒泡排序
int[] arr = {1, 5, 9, 7, 2, 3, 7, -1, 0, 3};
for (int i = 0; i < arr.length; i  ) {
    for (int j = 0; j < arr.length - i - 1; j  ) {
        if (arr[j] < arr[j   1]) {
            int temp = arr[j];
            arr[j] = arr[j   1];
            arr[j   1] = temp;
        }
    }
}

可以发现以上几种方法，我们都需要写很多代码，无法将注意力集中在设计自定义排序这个问题上，但是通过使用stream，我们就可以写出下面这样简洁的代码(如果愿意的话，你也可以把一系列的链式操作写在一行上，但为了代码的可读性，不建议那么做)：

代码语言：javascript复制

int[] arr = {1, 5, 9, 7, 2, 3, 7, -1, 0, 3};
arr = IntStream.of(arr)
               .boxed()
               .sorted(Comparator.reverseOrder())
               .mapToInt(Integer::intValue)
               .toArray();

注意

在这里其实为了实现数组的逆序，我们只需要调用Arrays的sort方法，然后再进行数组元素的反转即可，不过因为是为了讲解自定义排序，大多数情况下不会是数组逆序这么简单，所以我就写了更加通用一些的代码。

统计数组中前 k 个个高频元素

在最后，我们通过一道题来进行实战以便更好的体验stream的强大之处，当然我们在练习该题的时候，更需要从算法的角度去考虑该题的解法，不过在本文，我们主要为了讲解stream的使用，所以就不去考虑算法的东西了，而如果使用stream，我们就可以写出下面这样简单易懂的代码：

代码语言：javascript复制

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        return Arrays.stream(nums)
                     .boxed()
                     .collect(Collectors.toMap(e -> e, e -> 1, Integer::sum))
                     .entrySet()
                     .stream()
                     .sorted((m1, m2) -> m2.getValue() - m1.getValue())
                     .limit(k)
                     .mapToInt(Map.Entry::getKey)
                     .toArray();
    }
}

基于 Spring Cloud Alibaba Gateway Nacos RocketMQ Vue & Element 实现的后台管理系统 用户小程序，支持 RBAC 动态权限、多租户、数据权限、工作流、三方登录、支付、短信、商城等功能

• 项目地址：https://gitee.com/zhijiantianya/yudao-cloud
• 视频教程：https://doc.iocoder.cn/video/

总结

本文介绍了几个简单、实用的stream使用技巧，当然stream的应用远不止此，希望通过本文，能够激发起你学习stream的兴趣，本文若有错误之处，也欢迎你的指正。

---

---

欢迎加入我的知识星球，一起探讨架构，交流源码。加入方式，长按下方二维码噢：

已在知识星球更新源码解析如下：

最近更新《芋道 SpringBoot 2.X 入门》系列，已经 101 余篇，覆盖了 MyBatis、Redis、MongoDB、ES、分库分表、读写分离、SpringMVC、Webflux、权限、WebSocket、Dubbo、RabbitMQ、RocketMQ、Kafka、性能测试等等内容。

提供近 3W 行代码的 SpringBoot 示例，以及超 4W 行代码的电商微服务项目。

获取方式：点“在看”，关注公众号并回复 666 领取，更多内容陆续奉上。

代码语言：javascript复制

文章有帮助的话，在看，转发吧。谢谢支持哟 (*^__^*）

编程算法 spring https 网络安全 apache

0 人点赞