java使用Stream流對自定義對象數(shù)組去重的實現(xiàn)

更新時間：2025年04月16日 10:54:35 作者：Aaron?Fang

本文主要介紹了java使用Stream流對自定義對象數(shù)組去重的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧

方法 1：直接使用 distinct()（需重寫 equals 和 hashCode）

若自定義對象已正確重寫 equals() 和 hashCode() 方法，可直接通過 distinct() 去重。
適用場景：對象的唯一性由所有字段共同決定（如數(shù)據(jù)庫實體類的主鍵）。

示例代碼

public class Person {
    private String id;
    private String name;

    // 構造方法、Getter/Setter 省略

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        Person person = (Person) o;
        return Objects.equals(id, person.id); // 根據(jù) id 判斷是否相等
    }

    @Override
    public int hashCode() {
        return Objects.hash(id); // 基于 id 生成哈希
    }
}

// 使用 Stream 去重
Person[] people = ...; // 自定義對象數(shù)組
List<Person> uniqueList = Arrays.stream(people)
        .distinct()
        .collect(Collectors.toList());

方法 2：基于對象的某個唯一屬性去重（無需重寫 equals 和 hashCode）

若無法修改對象類（如第三方庫的類），或需根據(jù)部分字段去重，可用 Collectors.toMap 或 TreeSet 實現(xiàn)。

(1) 使用 Collectors.toMap

適用場景：根據(jù)唯一鍵（如 id）去重，保留第一個出現(xiàn)的元素。

List<Person> uniqueList = Arrays.stream(people)
        .collect(Collectors.toMap(
                Person::getId, // Key 提取函數(shù)（根據(jù) id 去重）
                p -> p,       // Value 為對象本身
                (existing, replacement) -> existing // 沖突時保留已存在的元素
        ))
        .values()            // 獲取去重后的 Value 集合
        .stream()
        .collect(Collectors.toList());

(2) 使用 TreeSet 自定義比較器

適用場景：需要根據(jù)多個字段去重，或動態(tài)指定去重規(guī)則。

List<Person> uniqueList = Arrays.stream(people)
        .collect(Collectors.toCollection(
            () -> new TreeSet<>(Comparator.comparing(p -> p.getId() + p.getName()))
        ))
        .stream()
        .collect(Collectors.toList());

方法 3：使用 filter + 內存狀態(tài)去重

適用場景：需根據(jù)動態(tài)條件去重（如去重后保留最后一個元素）。

(1) 使用 ConcurrentHashMap 維護狀態(tài)

Set<String> seenIds = ConcurrentHashMap.newKeySet();
List<Person> uniqueList = Arrays.stream(people)
        .filter(p -> seenIds.add(p.getId())) // 若 id 未出現(xiàn)過，保留
        .collect(Collectors.toList());

(2) 保留最后一個出現(xiàn)的元素

List<Person> uniqueList = Arrays.stream(people)
        .collect(Collectors.toMap(
                Person::getId,
                p -> p,
                (oldValue, newValue) -> newValue // 沖突時保留新元素
        ))
        .values()
        .stream()
        .collect(Collectors.toList());