java 优化双重for循环

首先我们要有两个对象分别是 学生信息 和 学生住宿信息

    class Student {
        //学生id
        private Integer studentUserId;

        //学生名称
        private String name;

    }

    class studentRooms {
        //学生id
        private Integer studentUserId;

        //学生名称
        private String roomInfo;

    }

将对象填充参数

        //学生信息
        List<Student> students = new ArrayList<>();
        students.add(Student.builder().studentUserId(111).name("张三").build());
        students.add(Student.builder().studentUserId(222).name("李四").build());

        //学生住宿信息
        List<StudentRoom> studentRooms = new ArrayList<>();
        studentRooms.add(StudentRoom.builder().studentUserId(222).roomInfo("住宿").build());
        studentRooms.add(StudentRoom.builder().studentUserId(111).roomInfo("走读").build());

这时我们的需求是, 将两边集合数据中, 相同studentUserId 数据拼接到一起返回给前端 如:

{
	"studentUserId":111,
	"name":"张三",
	"roomInfo":"走读"
}

这时候我们常规的写法就是 遍历两个集合 判断studentUserId 如果相同则拼装数据

for (Student student : students) {
    for (StudentRoom studentRoom : studentRooms) {
         //判断学生id是否相同
         if (student.getStudentUserId().equals(studentRoom.getStudentUserId())) {
            System.out.println("拼装数据返回给前端");
         }
     }
}

上述这种写法面临最大的问题就是时间复杂度为O(n^2)对 n 个数进行排序，需要扫描 n X n 次
对于上述代码也就是 2 x 2 = 4次

那么如何优化这种情况呢? 这里我们可以采用JDK1.8新特性的 stream流做处理代码如下

      //根据学生Id转成Map
      Map<Integer, StudentRoom> studentRoomByUserId = studentRooms.stream().collect(Collectors.toMap(StudentRoom::getStudentUserId, Function.identity()));
      for (Student student : students) {
            //根据外层遍历的学信息id get学生住宿信息Map中的Key
            StudentRoom studentRoom = studentRoomByUserId.get(student.getStudentUserId());
            //如果能get到数据!=null 说明id一样
            if (studentRoom != null) {
                System.out.println("拼装数据返回给前端");
            }
        }

按照上述这种操作方法会大大减少时间复杂度为O(n), 而循环的次数也变为了2次
完整代码

public class Test{

    @Data
    @Builder
    class Student {
        //学生id
        private Integer studentUserId;

        //学生名称
        private String name;

    }

    @Data
    @Builder
    class StudentRoom {
        //学生id
        private Integer studentUserId;

        //学生住宿信息
        private String roomInfo;
    }

    public void test() {

        //学生信息
        List<Student> students = new ArrayList<>();
        students.add(Student.builder().studentUserId(111).name("张三").build());
        students.add(Student.builder().studentUserId(222).name("李四").build());

        //学生住宿信息
        List<StudentRoom> studentRooms = new ArrayList<>();
        studentRooms.add(StudentRoom.builder().studentUserId(222).roomInfo("住宿").build());
        studentRooms.add(StudentRoom.builder().studentUserId(111).roomInfo("走读").build());

        //根据学生Id转成Map
        Map<Integer, StudentRoom> studentRoomByUserId = studentRooms.stream().collect(Collectors.toMap(StudentRoom::getStudentUserId, Function.identity()));
        
        for (Student student : students) {
            //根据外层遍历的学信息id get学生住宿信息Map中的Key
            StudentRoom studentRoom = studentRoomByUserId.get(student.getStudentUserId());
            //如果能get到数据说明id一样
            if (studentRoom != null) {
                System.out.println("拼装数据返回给前端");
            }
        }
    }


    //学生信息List

}