Java之HashMap迭代删除使用方法小结

栏目: Java · 发布时间: 7年前

内容简介：map的迭代删除，和我们常见的list，set不太一样，不能直接获取Iteraotr对象，提供的删除方法也是单个的，根据key进行删除，如果我们有个需求，将map中满足某些条件的元素删除掉，要怎么做呢？迭代删除，在不考虑并发安全的前提下，我们看下可以怎么支持我们知道map并不是继承自Collection接口的，HashMap 也没有提供迭代支持，既然没法直接迭代，那我就老老实的low b版好了

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);