Java Map Remove详解

QKTNT • 2025-04-18 13:40 • 编程

一、Map.remove()

Map.remove(Object key)方法用于根据key删除Map中的entry。如果Map中不存在这个key，则不进行任何操作。remove()方法在执行删除操作后会返回被删除的value值，或是null。

Map map = new HashMap();
map.put(1, "a");
String value = map.remove(1);
System.out.println(value); // 输出a
System.out.println(map.containsKey(1)); // 输出false

上面代码中，通过remove()方法删除了key为1的entry，并返回了对应的value值”a”。使用containsKey()方法查看发现在Map中已经不存在key为1的entry，返回false。

二、Map.remove()用法小技巧

1. 使用remove()方法批量删除符合条件entries

remove()方法不仅可以删除单个entry，还可以用于批量删除符合条件的entries。可以通过循环遍历或Java 8中的stream()方法实现。

// 循环遍历删除
Map map = new HashMap();
for (Iterator<Map.Entry> it = map.entrySet().iterator(); it.hasNext(); ) {
    Map.Entry entry = it.next();
    if (entry.getValue() == 0) {
        it.remove();
    }
}

// Java 8中的stream()方法实现
Map map = new HashMap();
map.entrySet().removeIf(entry -> entry.getValue() == 0);

上述代码中，使用removeIf()方法批量删除value为0的entry。

2. 使用remove()方法实现LRU缓存淘汰策略

LRU（Least Recently Used）缓存淘汰策略是指在缓存容量不够时，优先删除使用时间距离当前时间最久的entry。可以使用LinkedHashMap实现LRU缓存淘汰策略，LinkedHashMap继承自HashMap，内部是由一个linked list实现的，它保留了entry插入的顺序。

public class LRUCache extends LinkedHashMap {
    private final int maxEntries;
    public LRUCache(int maxEntries) {
        super(16, 0.75f, true);
        this.maxEntries = maxEntries;
    }
    // LinkedHashMap自带的删除最旧entry的办法，被覆盖时会调用这个方法
    protected boolean removeEldestEntry(Map.Entry eldest) {
        return size() > maxEntries;
    }
}

// 测试使用
LRUCache map = new LRUCache(2);
map.put(1, 1);
map.put(2, 2);
map.put(3, 3);
map.put(4, 4);
System.out.println(map);

在上述代码中，定义了一个LRUCache类，它继承自LinkedHashMap，并覆盖了removeEldestEntry()方法，实现LRU缓存淘汰策略。在测试使用时，map容量为2，但是存放了4个entry，因此会自动删除最久未使用的entry。

三、注意事项

1. Java集合类本身不是线程安全的

Java中的集合类本身是不具备线程安全性的，因为多个线程同时修改集合中的entry时，存在同时读写集合的问题。因此，在并发环境中使用集合类时，需要通过加锁或并发包提供的类来保证线程安全。

2. 在遍历集合时，不要使用循环表达式中的remove()方法

使用循环表达式中的remove()方法删除entry会导致遍历集合出现异常或遗漏元素。因为在使用remove()方法删除entry后，下一个元素会顶替删除的元素，因此遍历时可能会跳过顶替位置变化后的元素。

最后，上面提到的LRU缓存淘汰策略在Java 8中还可以通过Java 8提供的LinkedHashMap的一些新特性很好地实现，感兴趣的可以自己去了解。

原创文章，作者：QKTNT，如若转载，请注明出处：https://www.506064.com/n/370045.html