java根据集合里某个字段去重

原创

mob649e815a6b81 2023-07-14 11:29:28 ©著作权

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

Java根据集合里某个字段去重

在Java编程中，我们经常需要对集合进行去重操作。有时候，我们只需要根据集合中某个字段的值进行去重，而不是完全相等的元素。本文将介绍如何使用Java的集合和Lambda表达式来实现根据集合中某个字段去重的操作。

问题背景

假设我们有一个包含学生对象的集合，每个学生对象包含姓名和年龄两个字段。我们希望根据学生对象的姓名字段去除重复的学生对象，即对于同名的学生对象，只保留一个。

public class Student {
    private String name;
    private int age;

    // 构造方法和访问器方法省略

    // getters and setters 省略
}

List<Student> students = new ArrayList<>();
students.add(new Student("Alice", 18));
students.add(new Student("Bob", 20));
students.add(new Student("Alice", 19));
students.add(new Student("Bob", 19));
students.add(new Student("Charlie", 21));
students.add(new Student("Bob", 20));

在上述示例中，我们有一个包含6个学生对象的集合。其中，有两个学生的姓名是"Alice"，两个学生的姓名是"Bob"，而其他学生的姓名都是不同的。

我们期望的结果是，只保留一个"Alice"和一个"Bob"的学生对象，而其他学生对象都被去除。

解决方案

为了根据集合中某个字段去重，我们可以使用Java 8引入的Lambda表达式和Stream API。首先，我们需要给Student类添加重写equals和hashCode方法的实现，以便集合可以正确比较和去重对象。

public class Student {
    private String name;
    private int age;

    // 构造方法和访问器方法省略

    // getters and setters 省略

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Student student = (Student) o;
        return Objects.equals(name, student.name);
    }

    @Override
    public int hashCode() {
        return Objects.hash(name);
    }
}

接下来，我们可以使用distinct方法配合Lambda表达式来去除重复的学生对象。distinct方法会返回一个去重后的流，我们可以通过调用collect方法将流转换为列表。

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

List<Student> distinctStudents = students.stream()
        .distinct()
        .collect(Collectors.toList());

在上述代码中，我们调用stream方法将集合转换为流，然后调用distinct方法去除重复的学生对象。最后，我们使用collect方法将去重后的流转换为列表。

如果我们只想根据姓名字段去除重复的学生对象，可以使用Collectors.toMap方法来实现自定义的去重操作。

List<Student> distinctStudents = students.stream()
        .collect(Collectors.toMap(Student::getName, student -> student, (s1, s2) -> s1))
        .values()
        .stream()
        .collect(Collectors.toList());

在上述代码中，我们调用toMap方法，将学生对象的姓名作为键，学生对象本身作为值。当遇到重复的键时，我们使用lambda表达式来决定保留哪个学生对象。最后，我们通过调用values方法获取去重后的值视图，并再次使用collect方法将流转换为列表。