Java中List的retainAll方法使用方法

小蓝 • 2024-12-13 13:31 • 编程

一、背景介绍

在Java中，List是一种经常使用的集合类型，它提供了一系列常用的操作方法。其中retainAll方法可以用来求两个集合（可以是List、Set、Collection等）的交集。

retainAll方法的定义为：

boolean retainAll(Collection c);

其中c表示要求交集的集合，方法返回一个boolean值，表示是否修改了当前集合（即是否有元素被删除了）。

二、retainAll方法的使用

下面我们以两个List集合为例，来介绍retainAll方法的使用方法。

1. 求两个List集合的交集

首先，我们需要创建两个List集合，分别为list1和list2。它们的元素类型可以是任意类型，比如String、Integer等。

List<String> list1 = new ArrayList<>();
list1.add("apple");
list1.add("banana");
list1.add("cherry");
list1.add("date");

List<String> list2 = new ArrayList<>();
list2.add("banana");
list2.add("date");
list2.add("elderberry");
list2.add("fig");

现在，我们想要求list1和list2的交集，可以使用retainAll方法，代码如下：

boolean isModified = list1.retainAll(list2);

执行完上面的代码后，list1中的元素将只剩下”banana”和”date”，而isModified的值为true，表示list1被修改了，即有元素被删除了。

2.使用retainAll方法去重

除了求交集之外，retainAll方法还可以用来去重。我们可以将一个List集合看做一个集合，然后将另一个List集合作为候选元素集合，使用retainAll方法求出它们的交集，再将交集添加到原List集合中，从而实现去重。

示例代码如下：

List<String> list1 = new ArrayList<>();
list1.add("apple");
list1.add("banana");
list1.add("cherry");

List<String> list2 = new ArrayList<>();
list2.add("banana");
list2.add("cherry");
list2.add("date");

boolean isModified = list1.retainAll(list2);
if (isModified) {
    list1.addAll(list2);
}

System.out.println(list1);

执行完上面的代码后，list1中的元素将为”apple”、”banana”、”cherry”和”date”，即去掉了重复的元素”banana”和”cherry”。

3. 针对自定义类型的使用方法

retainAll方法不仅可以用来处理字符串、整数等基本类型的List集合，还可以处理自定义类型的List集合。我们只需要在自定义类型中实现equals方法，即可确保retainAll方法可以正确比较元素的相等性。

示例代码如下：

class Person {
    private String name;
    private int age;

    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    @Override
    public boolean equals(Object obj) {
        if (obj == null) {
            return false;
        }
        if (this == obj) {
            return true;
        }
        if (obj instanceof Person) {
            Person person = (Person) obj;
            return Objects.equals(this.name, person.name) && this.age == person.age;
        }
        return false;
    }
}

public class Main {

    public static void main(String[] args) {
        List<Person> list1 = new ArrayList<>();
        list1.add(new Person("Alice", 18));
        list1.add(new Person("Bob", 20));
        list1.add(new Person("Charlie", 22));

        List<Person> list2 = new ArrayList<>();
        list2.add(new Person("Bob", 20));
        list2.add(new Person("Charlie", 22));
        list2.add(new Person("David", 24));

        boolean isModified = list1.retainAll(list2);
        if (isModified) {
            list1.addAll(list2);
        }

        list1.forEach(person -> System.out.println(person.getName() + " " + person.getAge()));
    }

}

执行完上面的代码后，输出结果为：

Bob 20
Charlie 22
David 24

可以看到，我们成功地对自定义类型的List集合求了交集，它们的equals方法正确比较了两个元素的相等性。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/250944.html