面试官:怎么删除 HashMap 中的元素?我一行代码搞定,赶紧拿去用!

简介: 面试官:怎么删除 HashMap 中的元素?我一行代码搞定,赶紧拿去用!

这篇就分享下如何删除 HashMap 中的元素吧!

PS: 这仅是我个人掌握的实现方案,不一定全,也不一定是最优的,欢迎大家分享,杠精勿扰。

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 示例都会在这上面提供!


一般删除 HashMap 集合中的元素,如果知道具体的 Key,并且需要根据 Key 删除元素,使用 remove 方法就可以了。但是如何根据 Value 删除 HashMap 集合中的元素呢?这才是你必须掌握的技巧!


1、使用 for 循环删除

/**
 * 使用 for 循环删除
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void remove1() {
    Set<Map.Entry<String, String>> entries = new CopyOnWriteArraySet<>(initMap.entrySet());
    for (Map.Entry<String, String> entry : entries) {
        if ("张三".equals(entry.getValue())) {
            initMap.remove(entry.getKey());
        }
    }
    System.out.println(initMap);
}

输出结果:


{user2=李四, user7=李四, user8=王五, user5=王五, user6=赵六, user4=李四}


使用 HashMap 中实现的 entrySet 方法获取元素的集合,然后再进行循环遍历,先根据 Value 值判断要删除的元素,然后再根据 Key 删除元素。


在之前的文章中知道,增强的 for 循环底层使用的迭代器 Iterator,而 HashMap 是 fail-fast 类型的错误机制,所以遍历时删除元素会出现 java.util.ConcurrentModificationException 并发修改异常。


所以,这里使用了线程安全的 CopyOnWriteArraySet 封装了一层,避免出现并发修改异常,java.util.concurrent 包中的并发集合类都被设计为 fail-safe(安全失败)类型的,比如 CopyOnWrite* 、ConcerrentHashMap 集合,遍历过程中结构发生变更是安全的,不会抛出以上异常。


需要注意的是:


虽然 CopyOnWriteArraySet 并发性能很好,但每次删除时都会复制一份同等集合,所以要考虑数据过多可能导致的内存消耗问题。具体使用和实现原理可以点击该 CopyOnWriteArraySet 关键字链接看之前的文章,这里不再撰述。


2、使用 forEach 循环删除

/**
 * 使用 forEach 循环删除
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void remove2() {
    ConcurrentHashMap<String, String> map = new ConcurrentHashMap<>(initMap);
    map.forEach((k, v) -> {
        if ("张三".equals(v)) {
            map.remove(k);
        }
    });
    System.out.println(map);
}



输出结果:


{user2=李四, user7=李四, user8=王五, user5=王五, user6=赵六, user4=李四}


使用 HashMap 自带的 forEach 循环删除指定值的元素,这里为什么使用线程安全的 ConcurrentHashMap 集合包装了一层,同样是为了避免并发修改异常。ConcurrentHashMap 在各版本中都使用了最优的锁设计方案,它的并发性能也是非常优异的。


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


3、使用 Iterator 迭代器删除

/**
 * 使用 Iterator 迭代器删除
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void remove3() {
    Iterator<Map.Entry<String, String>> iterator = initMap.entrySet().iterator();
    while (iterator.hasNext()) {
        Map.Entry<String, String> entry = iterator.next();
        if ("张三".equals(entry.getValue())) {
            iterator.remove();
        }
    }
    System.out.println(initMap);
}




输出结果:


{user2=李四, user7=李四, user8=王五, user5=王五, user6=赵六, user4=李四}


这种方式即正常使用迭代器遍历删除,它不会发生并发修改异常。


需要注意的是:


这种方法虽然不会发生并发修改异常,但 HashMap 并不是线程安全的,在迭代删除元素时,另一个线程可能会删除 HashMap 中的数据, 这时使用迭代器删除同样会导致并发修改异常。


所以,要保证线程安全的删除,在创建迭代器之前,可以先用线程安全的 ConcurrentHashMap 集合包装一层。或者使用 synchronized 关键字锁住整个 Map。


如果没有多线程修改环境,可以不用考虑。


4、使用 removeIf 删除

/**
 * 使用 removeIf 删除
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void remove4() {
    initMap.entrySet().removeIf(entry -> "张三".equals(entry.getValue()));
    System.out.println(initMap);
}



输出结果:


{user2=李四, user7=李四, user8=王五, user5=王五, user6=赵六, user4=李四}


使用 entrySet 的 removeIf 删除,它底层使用的是迭代器:


default boolean removeIf(Predicate<? super E> filter) {
    Objects.requireNonNull(filter);
    boolean removed = false;
    final Iterator<E> each = iterator();
    while (each.hasNext()) {
        if (filter.test(each.next())) {
            each.remove();
            removed = true;
        }
    }
    return removed;
}



所以,它和方法 3 是一样的,只不过把条件写成了 Predicate 函数式接口而已。


需要注意的是:


removeIf 虽然更方便了,但它仍然不是线程安全的,多线程场景参考方案同方法 3。


5、使用 Stream 删除

/**
 * 使用 Stream 删除
 * @author: 栈长
 * @from: 公众号Java技术栈
 */
@Test
public void remove5() {
    Map<String, String> map = initMap.entrySet().stream()
            .filter(entry -> !"张三".equals(entry.getValue()))
            .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
    System.out.println(map);
}



输出结果:


{user2=李四, user7=李四, user8=王五, user5=王五, user6=赵六, user4=李四}


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


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


https://github.com/javastacks/javastack


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


总结

本文总结了 5 种删除 HashMap 元素的方法:


  • 使用 for 循环删除
  • 使用 forEach 循环删除
  • 使用 Iterator 迭代器删除
  • 使用 removeIf 删除
  • 使用 Stream 删除

实际开发过程中,可能会使用不同的遍历方式,所以重点要考虑多线程场景,如果只是简单的删除元素,使用 removeIf 和 Stream 过滤是最省事的。


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




相关文章
|
13天前
|
存储 算法 Java
Java查找算法概览:二分查找适用于有序数组,通过比较中间元素缩小搜索范围;哈希查找利用哈希函数快速定位,示例中使用HashMap存储键值对,支持多值关联。
【6月更文挑战第21天】Java查找算法概览:二分查找适用于有序数组,通过比较中间元素缩小搜索范围;哈希查找利用哈希函数快速定位,示例中使用HashMap存储键值对,支持多值关联。简单哈希表实现未涵盖冲突解决和删除操作。
18 1
|
2月前
|
SQL 分布式计算 算法
2024年最新【Python】列表元素的 删除 操作(remove()、pop()、切片,2024年最新Python社招面试题
2024年最新【Python】列表元素的 删除 操作(remove()、pop()、切片,2024年最新Python社招面试题
2024年最新【Python】列表元素的 删除 操作(remove()、pop()、切片,2024年最新Python社招面试题
|
26天前
|
消息中间件 存储 缓存
面试题--HashMap和TreeMap的区别和应用场景有啥区别?
然后底层调用key的hashCode()方法得出hash值; 过哈希表哈希算法,将hash值转换成数组的下标(注1),下标位置上如果没有任何元素,就把Node添加到这个位置上。如果说下标对应的位置上有值。此时,就会拿着key和链表上每个节点的key进行equal。如果所有的equals方法返回都是false,那么这个新的节点将被添加到链表的末尾。如其中有一个equals返回了true,那么这个节点的value将会被覆盖,如果最终长度大于8就会转成红黑树,红黑树插入;
22 3
|
7天前
|
存储 安全 Java
《ArrayList & HashMap 源码类基础面试题》面试官们最喜欢问的ArrayList & HashMap源码类初级问,你都会了?
《ArrayList & HashMap 源码类基础面试题》面试官们最喜欢问的ArrayList & HashMap源码类初级问,你都会了?
11 0
|
7天前
【年前最后一波装逼】记一次阿里面试,我是如何用一行代码解决约瑟夫环问题的
【年前最后一波装逼】记一次阿里面试,我是如何用一行代码解决约瑟夫环问题的
|
14天前
|
存储 算法 Java
面试高频算法题汇总「图文解析 + 教学视频 + 范例代码」之 二分 + 哈希表 + 堆 + 优先队列 合集
面试高频算法题汇总「图文解析 + 教学视频 + 范例代码」之 二分 + 哈希表 + 堆 + 优先队列 合集
|
16天前
|
存储 安全 Java
Java Map新玩法:探索HashMap和TreeMap的高级特性,让你的代码更强大!
【6月更文挑战第18天】在Java中,Map的两大代表HashMap和TreeMap提供高效键值对操作。HashMap允许设定初始容量和加载因子以优化性能,非线程安全,可借助synchronized或ConcurrentHashMap处理多线程。遍历可通过entrySet()、keySet()和values()。而TreeMap依据键进行排序,支持自然排序和自定义Comparator,提供范围查询、获取首尾键的功能,适用于需有序遍历的场景。理解和利用这些特性能增强代码功能和效率。
|
2月前
|
数据采集 数据挖掘 Python
最全妙不可言。写出优雅的 Python 代码的七条重要技巧,2024年最新被面试官怼了还有戏吗
最全妙不可言。写出优雅的 Python 代码的七条重要技巧,2024年最新被面试官怼了还有戏吗
|
2月前
|
数据采集 XML 程序员
最新用Python做垃圾分类_python垃圾分类代码用key和format,5年经验Python程序员面试27天
最新用Python做垃圾分类_python垃圾分类代码用key和format,5年经验Python程序员面试27天
最新用Python做垃圾分类_python垃圾分类代码用key和format,5年经验Python程序员面试27天
|
2月前
|
数据采集 机器学习/深度学习 人工智能
最新用python代码画爱心,来自程序猿的浪漫~_python画爱心代码(1),2024年最新面试简历模板免费
最新用python代码画爱心,来自程序猿的浪漫~_python画爱心代码(1),2024年最新面试简历模板免费
最新用python代码画爱心,来自程序猿的浪漫~_python画爱心代码(1),2024年最新面试简历模板免费