Java8单迭代实现ID匹配为EmployeeList添加PinCode

核心思路 #

要避免O(n²)的时间复杂度，关键是先将AddressList转换为以员工ID为键、PinCode为值的HashMap，这样根据ID查找PinCode的操作是O(1)。之后只需单次遍历EmployeeList，通过Map快速匹配赋值，整体时间复杂度为O(n+m)（n为员工列表长度，m为地址列表长度）。

假设实体类定义 #

先给出示例实体类（如果你的类结构不同，只需调整对应的getter/setter即可）：

// Employee类
public class Employee {
    private String id;
    private String name;
    private String pinCode; // 需要被赋值的字段

    // 构造器、getter和setter
    public String getId() { return id; }
    public void setPinCode(String pinCode) { this.pinCode = pinCode; }
    // 其他getter/setter省略
}

// Address类
public class Address {
    private String employeeId; // 与Employee.id匹配的关联字段
    private String pinCode;

    // 构造器、getter和setter
    public String getEmployeeId() { return employeeId; }
    public String getPinCode() { return pinCode; }
}

具体实现代码 #

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

public class EmployeeAddressMatcher {
    public static void main(String[] args) {
        List<Employee> employeeList = new ArrayList<>();
        // 此处填充你的Employee数据
        List<Address> addressList = new ArrayList<>();
        // 此处填充你的Address数据

        // 1. 将AddressList转换为ID-PinCode映射表
        Map<String, String> idToPinCodeMap = addressList.stream()
                .collect(Collectors.toMap(
                        Address::getEmployeeId,
                        Address::getPinCode,
                        (existingPin, newPin) -> existingPin // 处理重复ID：保留第一个匹配的PinCode
                ));

        // 2. 单次遍历EmployeeList，匹配赋值
        employeeList.forEach(employee -> {
            String matchedPinCode = idToPinCodeMap.get(employee.getId());
            if (matchedPinCode != null) {
                // 仅当找到匹配地址时才赋值，保留员工原有其他数据
                employee.setPinCode(matchedPinCode);
            }
        });
    }
}

关键细节说明 #

• 重复ID处理：如果AddressList中存在同一个员工ID对应多个地址的情况，Collectors.toMap的第三个参数可以指定冲突处理逻辑，示例中选择保留第一个出现的PinCode，你可以根据业务需求改为保留最后一个或合并值。
• 数据保留：遍历员工列表时，仅在找到匹配PinCode时才更新字段，未匹配到的员工，其pinCode字段会保持原有值（比如初始的null或默认值），完全保留原有数据。
• 性能优势：构建Map的时间复杂度为O(m)，遍历员工列表为O(n)，总时间复杂度远低于嵌套循环的O(n*m)，数据量越大，性能提升越明显。

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