java list 拉姆达取交集

原创

mob64ca12df277e 2024-12-25 05:23:02 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12df277e的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java List 拉姆达取交集

在日常开发中，我们常常需要对集合进行各种操作，其中取交集的操作是非常常见的。在Java中，传统的方式是通过循环遍历两个集合并找出交集。然而，使用Java 8及以后的版本引入的Lambda表达式和流（Stream）API，我们可以以更简洁的方式来处理这种情况。本文将详细介绍如何使用Lambda表达式来取两个List的交集，并提供相应的代码示例。

1. 什么是交集？

交集是指两个集合中共同包含的元素集合。例如，若集合A={1, 2, 3}与集合B={2, 3, 4}，则集合A与B的交集为{2, 3}。

2. 使用Lambda表达式进行交集操作

在Java中，我们可以通过Stream的filter方法结合collect方法来获取两个List的交集。首先，我们需要引入必要的包：

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

接下来，我们定义一个方法来计算交集：

public static List<Integer> getIntersection(List<Integer> listA, List<Integer> listB) {
    return listA.stream()
            .filter(listB::contains)
            .collect(Collectors.toList());
}

在这个方法中，我们将listA转换为一个Stream，使用filter方法找到在listB中也存在的元素，最后将结果收集为一个List。

3. 完整示例

以下是一个完整的Java程序示例，其中演示了如何使用上面定义的getIntersection方法来获取两个List的交集：

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

public class Main {
    public static void main(String[] args) {
        List<Integer> listA = Arrays.asList(1, 2, 3, 4, 5);
        List<Integer> listB = Arrays.asList(4, 5, 6, 7, 8);
        
        List<Integer> intersection = getIntersection(listA, listB);
        
        System.out.println("交集: " + intersection);
    }
    
    public static List<Integer> getIntersection(List<Integer> listA, List<Integer> listB) {
        return listA.stream()
                .filter(listB::contains)
                .collect(Collectors.toList());
    }
}

运行结果

当我们运行上面的程序时，输出结果将是：

交集: [4, 5]

4. Lambda表格展示

我们可以用一个表格展示不同List和其交集的示例：

List A	List B	Intersection
[1, 2, 3]	[2, 3, 4]	[2, 3]
[4, 5, 6]	[5, 6, 7]	[5, 6]
[10, 20, 30]	[20, 30, 40]	[20, 30]

5. 序列图示例

下面的序列图展示了getIntersection方法的调用过程：

sequenceDiagram
    participant A as List A
    participant B as List B
    participant C as getIntersection
    participant D as Result List

    A->>C: 调用getIntersection(listA, listB)
    C->>A: stream()
    C->>B: filter(listB::contains)
    C->>D: collect(Collectors.toList())
    D->>C: 返回交集
    C->>A: 输出交集