面试官:怎么删除 HashMap 中的重复元素?第 3 种实现思路,99% 的...

Java技术栈

共 5294字,需浏览 11分钟

 · 2023-03-13

点击关注公众号,Java干货 及时送达 e8ea5d6c2a9c4cd7922a875aa4149822.webp

推荐阅读:

劝大家别再搞老旧的 Spring Cloud 了!

Spring Cloud 2022 正式发布!



背景

大家好,我是栈长。

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

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 教程菜单中阅读。

Collectors.toMap 方法返回的是一个Collector,它可以将元素累积到 Map 中,Map 的键和值将提供的映射函数应用到输入元素的结果,如果映射的键包含重复项,则值映射函数会使用提供的 merge 函数进行结果合并。

Collectors.toMap 方法可以对 Key 进行去重合并,这也是为什么进行了两次 collect 收集的原因:

第一次收集:

把 Value 作为 Key,Key 作为 Value,这样就能使用 Value 进行去重了,输出结果:

{李四=user2, 张三=user1, 王五=user8, 赵六=user6}

虽然能去重了,但是 HashMap 中的 Key 和 Value 值却颠倒了,所以需要第二次收集。

第二次收集:

现在的 Key 是之前的 Value,所以需要再相互换过来,输出结果:

{user1=张三, user2=李四, user8=王五, user6=赵六}

这个方法比较绕,虽然能一行代码搞定,但代码很冗余,不是很优雅,最重要的是这两次的收集过程会创建两次新 Map,相对比较耗内存。

总结

本文总结了 3 种删除 HashMap 重复元素的方法:

  • 新创建 Map 添加不重复元素
  • 添加 Set 再删除重复元素(推荐)
  • 使用 Stream 删除重复元素

实际开发过程中,可能会使用不同的遍历方式,使用哪种删除方案可以根据不同的遍历方式进行选择,但推荐使用 Set 方案,可以直接删除 Map 中的重复元素,不会创建新的 HashMap。

另外,遍历集合时需要重点考虑是否有多线程修改元素的场景,可能导致的并发修改异常,参考之前文章中介绍的方案,这里不再撰述了。

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

https://github.com/javastacks/javastack

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

你身边还有谁不会删除 HashMap 中的重复元素?把这篇文章发给他吧,让大家少走弯路,少写垃圾代码,共同进步。

你还知道哪些 HashMap 去重技巧?欢迎留言分享~

好了,今天的分享就到这里了,后面栈长会分享更多好玩的 Java 技术和最新的技术资讯,关注公众号Java技术栈第一时间推送,我也将主流 Java 面试题和参考答案都整理好了,大家可以在Java面试库小程序进行刷题。

版权声明: 本文系公众号 "Java技术栈" 原创,转载、引用本文内容请注明出处,抄袭、洗稿一律投诉侵权,后果自负,并保留追究其法律责任的权利。

End


你还只会用 AtomicXXX ?!恭喜你,可以下岗了!

Spring Cloud Sleuth 被官方彻底移除!

在公司做了三年 SSH 老项目,才发现微服务真香!

怎么去除 List 中的重复元素?我一行代码搞定!

带了一个 3 年的开发,不会循环删除 List 中的元素

7fadc7f62f205fc586f44eafe8361c75.webpSpring Cloud 微服务最新课程!
浏览 27
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报