java 矩阵降维_Java和Kotlin 的降维方法

Java 和 kotlin 的降维方法

在很多时候都会遇到一个问题，一个集合中包含多个集合，但是又要统一添加到一个中，也就是俗称的降维(把二维数组降成一维数组)

前提(这里使用的lombok 省略了get、set、有参、无参和toString方法，没有的需要手动快捷键生成对应的方法哦)：

User.class

@Data

@AllArgsConstructor

@NoArgsConstructor

@ToString

public class User {

private Long id;

private String username;

}

jdk版本 1.8

idea 对用的快捷键：Alt + Insert

Java

Spring boot的测试类：

@SpringBootTest

class UserServerApplicationTests {

@Test

void contextLoads() {

List> lists = new ArrayList>();

for (int i = 0; i < 3; i++) {

List list = new ArrayList<>();

for (int j = 0; j < 3; j++) {

User user = new User();

user.setUsername("小黑呀" + i + "-" + j);

list.add(user);

}

lists.add(list);

}

//代码都是拆自这，下面的代码直接粘贴到这就可以了

}

}

调用方法前的参数：

[[User(id=null, username=小黑呀0-0), User(id=null, username=小黑呀0-1), User(id=null, username=小黑呀0-2)], [User(id=null, username=小黑呀1-0), User(id=null, username=小黑呀1-1), User(id=null, username=小黑呀1-2)], [User(id=null, username=小黑呀2-0), User(id=null, username=小黑呀2-1), User(id=null, username=小黑呀2-2)]]

方案有几种：普通for循环(不做了解)、forEach方法、Lambda 函数。

forEach循环：

普通试用版本(推荐)：

List userList = new ArrayList<>();

lists.forEach(it -> userList.addAll(it));

System.out.println(userList.toString());

修改一版(推荐)这里使用了jdk 1.8新增的 :: 语法，类似于方法引用：

List userList = new ArrayList<>();

lists.forEach(userList::addAll);

System.out.println(userList.toString());

调用结果：

[User(id=null, username=小黑呀0-0), User(id=null, username=小黑呀0-1), User(id=null, username=小黑呀0-2), User(id=null, username=小黑呀1-0), User(id=null, username=小黑呀1-1), User(id=null, username=小黑呀1-2), User(id=null, username=小黑呀2-0), User(id=null, username=小黑呀2-1), User(id=null, username=小黑呀2-2)]

Lambda函数(这里使用flatMap方法)：

flatMap 是可以自定义方法的，当官方提供的不够用时就需要自己写逻辑

//这个就是对所有的一个降维方案：

List userList = lists.stream().

flatMap(Collection::stream).

collect(Collectors.toList());

System.out.println(userList.toString());

整个对象降维的结果为

[User(id=null, username=小黑呀0-0), User(id=null, username=小黑呀0-1), User(id=null, username=小黑呀0-2), User(id=null, username=小黑呀1-0), User(id=null, username=小黑呀1-1), User(id=null, username=小黑呀1-2), User(id=null, username=小黑呀2-0), User(id=null, username=小黑呀2-1), User(id=null, username=小黑呀2-2)]

//保不齐我只需要这个user对象的一个字段，比如我只需要一个用户的名称的list

List userNameList = lists.stream().

flatMap(it -> it.stream().map(User::getUsername)).

collect(Collectors.toList());

System.out.println(userNameList.toString());

降维但只获取某个字段的结果

[小黑呀0-0, 小黑呀0-1, 小黑呀0-2, 小黑呀1-0, 小黑呀1-1, 小黑呀1-2, 小黑呀2-0, 小黑呀2-1, 小黑呀2-2]

拓展：

如果需要返回结果去重请使用: Collectors.toSet()

循环输出也可以采用forEach: lists.forEach(System.out::println)

Kotlin

Spring boot Kotlin测试类

@SpringBootTest

class WebApplicationTests {

private val logger: Logger = LoggerFactory.getLogger(this::class.java)

@Test

fun t() {

val lists: MutableList> = ArrayList()

for (i in 0..2) {

val list: MutableList = ArrayList()

for (j in 0..3) {

val user = User()

user.username= "小黑呀${i}-${j}"

list.add(user)

}

lists.add(list)

}

//下面的代码同样是在这个部位移植出去的，粘贴回即可

}

}

调用方法前的参数展示：

[[User(id=null, username=小黑呀0-0), User(id=null, username=小黑呀0-1), User(id=null, username=小黑呀0-2)], [User(id=null, username=小黑呀1-0), User(id=null, username=小黑呀1-1), User(id=null, username=小黑呀1-2)], [User(id=null, username=小黑呀2-0), User(id=null, username=小黑呀2-1), User(id=null, username=小黑呀2-2)]]

kotlin 降维在他提供的函数中有两个满足要求，

1.flatten 这个方法不需要自己去写什么代码，直接调用该方法即可

2.flatMap 这个方法就需要自己写，但是需要针对返回某个字段的需求

使用flatten

val userList = lists.flatten()

println(userList.toString())

调用结果

[User(id=null, username=小黑呀0-0), User(id=null, username=小黑呀0-1), User(id=null, username=小黑呀0-2), User(id=null, username=小黑呀1-0), User(id=null, username=小黑呀1-1), User(id=null, username=小黑呀1-2), User(id=null, username=小黑呀2-0), User(id=null, username=小黑呀2-1), User(id=null, username=小黑呀2-2)]

使用flatMap(这个方法一般用于返回类中的字段)

val idList = lists.flatMap {it -> it.map { o -> o.username} }

println(idList.toString())

调用结果

[小黑呀0-0, 小黑呀0-1, 小黑呀0-2, 小黑呀1-0, 小黑呀1-1, 小黑呀1-2, 小黑呀2-0, 小黑呀2-1, 小黑呀2-2]