17. HashMap便利删除使用姿势

一灰灰blogJavaJDKJava约 444 字大约 1 分钟

map的迭代删除,和我们常见的list,set不太一样,不能直接获取Iteraotr对象,提供的删除方法也是单个的,根据key进行删除,如果我们有个需求,将map中满足某些条件的元素删除掉,要怎么做呢?

I. Map 迭代删除

迭代删除,在不考虑并发安全的前提下,我们看下可以怎么支持

1. 非常不优雅版本

我们知道map并不是继承自Collection接口的,HashMap 也没有提供迭代支持,既然没法直接迭代,那我就老老实的low b版好了

Map<String, Integer> map = new HashMap<>();
map.put("a", 1);
map.put("b", 2);
map.put("c", 3);
map.put("d", 4);

List<String> removeKey = new ArrayList<>();
for (Map.Entry<String, Integer> e: map.entrySet()) {
	if (e.getValue() % 2== 0) {
	    removeKey.add(e.getKey());
	}
}
removeKey.forEach(map::remove);

上面的实现怎么样?并没有什么毛病

(为啥不直接在遍历中删除?)

2. 正确姿势版

虽然Map没有迭代,但是它的entrySet有啊,所以我们可以通过它来实现遍历删除

Map<String, Integer> map = new HashMap<>();
map.put("a", 1);
map.put("b", 2);
map.put("c", 3);
map.put("d", 4);

Iterator<Map.Entry<String, Integer>> iterator = map.entrySet().iterator();
Map.Entry<String, Integer> entry;
while (iterator.hasNext()) {
    entry = iterator.next();
    if (entry.getValue() % 2 == 0) {
        iterator.remove();
    }
}
System.out.println(map);

上面这个可能是我们经常使用的操作姿势了,利用迭代器来操作元素

3. 简洁版

到jdk8之后,针对容器提供了很多简洁的操作方式,迭代删除这方面可以说更加简单了

Map<String, Integer> map = new HashMap<>();
map.put("a", 1);
map.put("b", 2);
map.put("c", 3);
map.put("d", 4);
map.entrySet().removeIf(entry -> entry.getValue() % 2 == 0);
Loading...