如何用Stream API高效实现两集合匹配并设置默认值？

当然有，通过预将集合B转换为以identifier为键的Map，结合Stream API可以把时间复杂度从嵌套循环的O(m*n)优化到O(m+n)（m是集合A的大小，n是集合B的大小），完全满足你的需求，具体实现如下：

1. 实体类示例 #

假设你的A、B类结构如下（根据实际情况调整getter）：

class A {
    private String id;
    private String foo;

    // 构造器、getter方法
    public String getId() { return id; }
    public String getFoo() { return foo; }
}

class B {
    private String identifier;
    private String bar;

    // 构造器、getter方法
    public String getIdentifier() { return identifier; }
    public String getBar() { return bar; }
}

2. 核心实现步骤 #

第一步：将集合B转换为快速查找的Map

把B集合转成以identifier为键的Map，后续匹配时可以O(1)时间找到对应元素：

List<B> bList = // 你的集合B实例
Map<String, B> bIdentifierMap = bList.stream()
    .collect(Collectors.toMap(B::getIdentifier, Function.identity()));

第二步：遍历集合A生成目标映射

利用Stream的Collectors.toMap构建最终结果，确保每个A元素都被保留，匹配不到时调用默认值函数：

List<A> aList = // 你的集合A实例
Map<A, String> resultMap = aList.stream()
    .collect(Collectors.toMap(
        // 键：直接使用A元素本身
        Function.identity(),
        // 值：匹配到B则拼接foo+bar，否则调用默认函数
        a -> Optional.ofNullable(bIdentifierMap.get(a.getId()))
            .map(b -> a.getFoo() + b.getBar())
            .orElseGet(() -> getDefaultValue(a)),
        // 处理键重复情况（如果A中有重复元素，保留第一个）
        (existingVal, newVal) -> existingVal
    ));

// 自定义默认值函数示例
private String getDefaultValue(A a) {
    // 根据业务逻辑生成默认值，比如返回a.getFoo() + "_default"
    return a.getFoo() + "_default";
}

方案优势 #

嵌套循环需要对每个A元素遍历整个B集合查找匹配，时间复杂度是O(m*n)；而先转Map后，每次查找都是O(1)的哈希表查询，整体时间复杂度降到O(m+n)，集合越大，性能提升越明显。

注意事项 #

• 如果A中存在逻辑相等的元素（即equals()返回true），需要根据业务需求调整合并函数(existingVal, newVal) -> existingVal，比如选择保留新值或合并值；如果A元素都是唯一的（基于对象引用或自定义equals），这个合并函数可以避免重复键异常。
• 如果identifier在B中存在重复，Collectors.toMap会抛出异常，此时可以改用重载方法指定合并规则，比如(oldB, newB) -> newB保留最后一个匹配的B元素。

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