Java如何依据指定name列表高效筛选对象列表中的匹配元素

核心优化思路 #

• 首先将List<String>类型的姓名匹配列表转换为HashSet<String>：HashSet的contains()方法时间复杂度为O(1)，远高于List的contains()的O(n)，整体筛选时间复杂度从O(m*n)降低到O(m+n)（m为姓名列表长度，n为对象列表长度），数据量越大性能优势越明显
• 转换HashSet的过程会自动对姓名列表去重，进一步降低匹配开销

代码实现 #

1. 自定义类A示例（POJO） #

public class A {
    private Long id;
    private String name;
    private String subject;

    // 构造方法、其他getter、setter此处省略
    public String getName() {
        return name;
    }
}

2. 核心筛选代码（Java 8+ Stream 写法，简洁高效） #

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

public class FilterDemo {
    public static void main(String[] args) {
        // 待筛选对象列表
        List<A> objectsA = new ArrayList<>();
        // 目标匹配姓名列表
        List<String> namesList = List.of("alpha","beta");

        // 第一步：将姓名列表转为HashSet，优化匹配效率
        Set<String> nameSet = new HashSet<>(namesList);

        // 第二步：筛选符合条件的对象，增加空值判断避免空指针
        List<A> filteredResult = objectsA.stream()
                .filter(a -> a.getName() != null && nameSet.contains(a.getName()))
                .collect(Collectors.toList());
    }
}

3. 兼容低版本Java的普通迭代写法 #

// 第一步同样先转HashSet
Set<String> nameSet = new HashSet<>(namesList);
List<A> filteredResult = new ArrayList<>();
for (A a : objectsA) {
    if (a.getName() != null && nameSet.contains(a.getName())) {
        filteredResult.add(a);
    }
}

特殊场景说明 #

• 如果姓名列表namesList本身长度极小（比如只有2、3个元素），直接用List的contains()性能差异可以忽略，但转HashSet的写法依然是通用最优解，兼容各种数据量场景
• 如果要求保留重复匹配的结果，不需要去重姓名，也依然推荐转HashSet，不影响匹配结果，只提升性能

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