面试官：怎么删除 HashMap 中的重复元素？第 3 种实现思路，99% 的人不会！

2023-03-14 09:29 由 Java技术栈发表于 #后端开发

背景

大家好，我是栈长。

前些天，栈长给大家分享了 3 篇实用的文章：

带了一个 3 年的开发，不会循环删除 List 中的元素，我简直崩溃！！
面试官：怎么去除 List 中的重复元素？我一行代码搞定，赶紧拿去用！
面试官：怎么删除 HashMap 中的元素？我一行代码搞定，赶紧拿去用！

List 和 Map 元素的删除、去重，这些都是工作中经常遇到的问题，一些基础程序员可能会走一些弯路，所以栈长输出了三篇，希望对大家有用，其中一些编程技巧很多老程序员也没用过，所以，技术真的是学无止境。

今天栈长带来集合的删除及去重系列的最后一篇，如何删除 HashMap 中的重复元素，即怎么根据 Value 去重，去除 HashMap 中 Value 重复的元素，这也是面试官可能会问到的。

为什么不是根据 Key 去重？

大家都知道，HashMap 的 key 是不会重复的，如果有重复就会用新值覆盖旧值。

当我们向一个 HashMap 中插入元素时，HashMap 会根据这个 key 的 equals 和 hashCode 方法进行判断，如果两个 key 的值用 equals 方法比较相同，且 key 的 hashCode 值也相同，那么 HashMap 将认为这是同一个 key，后续插入相同 key 的键值对会将旧值替换为新值。

需要注意的是：

Java 中的基本数据类型和 String 等内置类，它们已经正确实现了 equals 和 hashCode 方法，可以直接用作 HashMap 的 key，而不会导致重复的 key 出现。

如果我们使用自定义类的对象作为 HashMap 的 key，需要保证这个类正确实现了 equals 和 hashCode 方法，否则可能会出现插入 "重复 key" 的情况，正常情况下，这是不符合规范和逻辑的。

HashMap 删除重复元素方案

以下 HashMap 初始测试数据：

public Map<String, String> initMap = new HashMap<>() {{
    put("user1", "张三");
    put("user2", "李四");
    put("user3", "张三");
    put("user4", "李四");
    put("user5", "王五");
    put("user6", "赵六");
    put("user7", "李四");
    put("user8", "王五");
}};

本文所有完整示例源代码已经上传：

https://github.com/javastacks/javastack

欢迎 Star 学习，后面 Java 示例都会在这上面提供！

1、新创建 Map 添加不重复元素

/**
 * 新创建 Map 添加不重复元素
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void removeDuplicated1() {
    Map<String, String> map = new HashMap<>();
    initMap.forEach((k, v) -> {
        if (!map.containsValue(v)) {
            map.put(k, v);
        }
    });
    System.out.println(map);
}

这种方法很原始，通过创建一个新 HashMap，添加元素前进行判断，如果元素在新 HashMap 中不存在才进行添加。

2、添加 Set 再删除重复元素

/**
 * 添加 Set 再删除重复元素
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void removeDuplicated2() {
    Set<String> set = new HashSet<>();
    Iterator<Map.Entry<String, String>> iterator = initMap.entrySet().iterator();
    while (iterator.hasNext()) {
        Map.Entry<String, String> entry = iterator.next();
        if (!set.add(entry.getValue())) {
            iterator.remove();
        }
    }
    System.out.println(initMap);
}

这种方法和第一种方法大同小异，通过创建一个 HashSet，然后遍历 HashMap，因为 HashSet 是不允许重复元素的，所以，如果 HashSet 能添加元素说明元素没有重复，否则说明元素重复了，然后删除即可。

另外，HashSet、HashMap 的关系也是面试必问的，如果你近期准备面试跳槽，建议在Java面试库小程序在线刷题，涵盖 2000+ 道 Java 面试题，几乎覆盖了所有主流技术面试题。

3、使用 Stream 删除重复元素

/**
 * 使用 Stream 删除重复元素
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void removeDuplicated3() {
    Map<String, String> resultMap = initMap.entrySet().stream().collect(
            Collectors.toMap(Map.Entry::getValue, Map.Entry::getKey, (key1, key2) -> key1)
    ).entrySet().stream().collect(
            Collectors.toMap(Map.Entry::getValue, Map.Entry::getKey, (key1, key2) -> key1)
    );
    System.out.println(resultMap);
}

利用 Stream 的 collect 方法重新进行收集，这个方法也十分简单，一行代码搞定，为了可读性，文中代码进行了换行。Stream 基础就不介绍了，Stream 系列我之前写过一个专题了，不懂的关注公众号Java技术栈，然后在公众号 Java 教程菜单中阅读。