如何高效实现含共同companyId的不同类型列表的companyName匹配赋值

你的思路完全正确——将customerList转换为HashMap是处理这类关联匹配问题的最优性能方案，相比嵌套for循环，能大幅降低时间复杂度，尤其在数据量较大时优势明显。

方案原理 #

嵌套for循环的时间复杂度是O(n*m)（n为Customer列表长度，m为Employee列表长度），当两个列表各有1000条数据时，需要执行100万次匹配操作；而HashMap方案的时间复杂度是O(n+m)，仅需2000次操作，数据量越大性能差距越显著。

具体实现步骤 #

1. 构建CompanyId与CompanyName的映射表
   把customerList转换为以companyId为Key、companyName为Value的HashMap，实现O(1)时间复杂度的查询。

2. 遍历Employee列表完成赋值
   循环遍历employeeList，通过每个Employee的companyId从HashMap中快速获取对应的companyName并赋值。

代码示例 #

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

// 假设Customer和Employee类已实现getter/setter方法
public class CompanyNameMatcher {
    public void matchCompanyNames(List<Customer> customerList, List<Employee> employeeList) {
        // 构建companyId到companyName的映射，处理重复companyId时保留第一个
        Map<Long, String> companyIdNameMap = customerList.stream()
                .collect(Collectors.toMap(
                        Customer::getCompanyId,
                        Customer::getCompanyName,
                        (existingName, newName) -> existingName
                ));

        // 为Employee赋值companyName
        employeeList.forEach(employee -> {
            String matchedName = companyIdNameMap.get(employee.getCompanyId());
            if (matchedName != null) {
                employee.setCompanyName(matchedName);
            }
            // 若需处理无匹配的情况，可在此添加逻辑，比如设置默认值
            // else { employee.setCompanyName("未知公司"); }
        });
    }
}

注意事项 #

• 重复companyId处理：如果customerList中存在多个相同companyId的Customer，Collectors.toMap的第三个参数可指定冲突策略（比如保留最后一个、合并等），需根据业务需求调整。
• 空值处理：若Employee的companyId在Customer列表中无匹配，HashMap的get方法会返回null，可根据业务决定是否设置默认值。

内容的提问来源于stack exchange，提问作者Haidepzai