java代码-优化一段代码的时间复杂度从 O(n*m) 到 O(n+m)

码力斜杠哥

1. if (ObjectUtil.isNotEmpty(pageResult) && ObjectUtil.isNotEmpty(pageResult.getList())){
   
2.     List<Long> collect = pageResult.getList()
   
3.             .stream()
   
4.             .map(monitorRptRecordDO::getKeySupervisionUnit)
   
5.             .filter(StrUtil::isNotBlank)
   
6.             .filter(s -> s.trim().matches("\\d+"))
   
7.             .map(Long::valueOf)
   
8.             .toList();
   
9.     List<KeyUnitQueryDO> list = soilGroundwaterMainService.getSoilGroundwaterMainByIds(collect);
   
10.     for (monitorRptRecordDO monitorRptRecordDO : pageResult.getList()) {
    
11.         if (ObjectUtil.isNotEmpty(list)){
    
12.             List<KeyUnitQueryDO> list1 = list.stream().filter(x -> ObjectUtil.equal(x.getId().toString(), monitorRptRecordDO.getKeySupervisionUnit())).toList();
    
13.             if (!list1.isEmpty()){
    
14.                 monitorRptRecordDO.setKeyMonitorUnitName(list1.getFirst().getKeySupervisionUnit());
    
15.             }
    
16. 
    
17.         }
    
18.     }
    
19. 
    
20. }

复制代码

修改后：

1. if (ObjectUtil.isNotEmpty(pageResult) && ObjectUtil.isNotEmpty(pageResult.getList())) {
   
2. 
   
3.     // 收集 ID（O(n)）
   
4.     List<Long> collect = pageResult.getList().stream()
   
5.             .map(monitorRptRecordDO::getKeySupervisionUnit)
   
6.             .filter(StrUtil::isNotBlank)
   
7.             .filter(s -> s.trim().matches("\\d+"))
   
8.             .map(Long::valueOf)
   
9.             .toList();
   
10. 
    
11.     // 批量查询（O(m)）
    
12.     List<KeyUnitQueryDO> list = soilGroundwaterMainService.getSoilGroundwaterMainByIds(collect);
    
13. 
    
14.     // 关键优化：list → Map（O(m)）
    
15.     Map<String, KeyUnitQueryDO> map = list.stream()
    
16.             .collect(Collectors.toMap(
    
17.                     x -> x.getId().toString(),   // key
    
18.                     x -> x,                      // value
    
19.                     (a, b) -> a                  // 防止重复 key 报错
    
20.             ));
    
21. 
    
22.     // 直接 O(1) 查 Map（O(n)）
    
23.     for (monitorRptRecordDO monitorRptRecordDO : pageResult.getList()) {
    
24. 
    
25.         if (ObjectUtil.isNotEmpty(map)) {
    
26.             // 核心在这里 （O(1)）因为map的时间复杂度就是o(1)
    
27.             KeyUnitQueryDO keyUnit = map.get(monitorRptRecordDO.getKeySupervisionUnit());
    
28. 
    
29.             if (ObjectUtil.isNotEmpty(keyUnit)) {
    
30.                 monitorRptRecordDO.setKeyMonitorUnitName(
    
31.                         keyUnit.getKeySupervisionUnit()
    
32.                 );
    
33.             }
    
34.         }
    
35.     }
    
36. }

复制代码

小贾

学到了楼主😎