当前位置：首页 > news >正文

Java中Map循环安全的删除数据的4中方法

news 来源：原创 2025/5/21 17:38:59

文章目录

前言
一、使用Iterator删除
二、使用 removeIf（Java 8+）
三、遍历时记录需要删除的键（不推荐）
四、使用 Stream（Java 8+）
总结

前言

在 Java 中，遍历 HashMap 并删除数据时，直接使用 for-each 循环或 Iterator 的 remove 方法可能会导致 ConcurrentModificationException 异常。为了避免这个问题，需要使用 Iterator 来安全地删除数据。

以下是常见的4中删除方法

一、使用Iterator删除

这种事最安全的方式！也是最推荐的一种方法！

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

public class Main {
    public static void main(String[] args) {
        HashMap<String, Integer> map = new HashMap<>();
        map.put("A", 1);
        map.put("B", 2);
        map.put("C", 3);

        // 使用 Iterator 遍历并删除
        Iterator<Map.Entry<String, Integer>> iterator = map.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<String, Integer> entry = iterator.next();
            if (entry.getValue() == 2) { // 删除值为 2 的条目
                iterator.remove(); // 安全删除
            }
        }

        System.out.println(map); // 输出: {A=1, C=3}
    }
}

说明：

使用iterator.remove() 方法删除当前的数据，并不是直接操作HahsMap
这种方法不会引发ConcurrentModificationException 报错

二、使用 removeIf（Java 8+）

如果你使用的是 Java 8 或更高版本，可以使用 removeIf 方法，这是一种更简洁的方式。

import java.util.HashMap;

public class Main {
    public static void main(String[] args) {
        HashMap<String, Integer> map = new HashMap<>();
        map.put("A", 1);
        map.put("B", 2);
        map.put("C", 3);

        // 使用 removeIf 删除符合条件的条目
        map.entrySet().removeIf(entry -> entry.getValue() == 2);

        System.out.println(map); // 输出: {A=1, C=3}
    }
}

说明：

removeIf 是 Collection 接口的方法，HashMap 的 entrySet() 返回一个集合，因此可以直接使用。
代码更简洁，适合 Java 8 及以上版本。

三、遍历时记录需要删除的键（不推荐）

如果你不想使用 Iterator 或 removeIf，可以先记录需要删除的键，然后再统一删除。这种方式不推荐，因为需要额外的空间。

import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;

public class Main {
    public static void main(String[] args) {
        HashMap<String, Integer> map = new HashMap<>();
        map.put("A", 1);
        map.put("B", 2);
        map.put("C", 3);

        // 记录需要删除的键
        Set<String> keysToRemove = new HashSet<>();
        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            if (entry.getValue() == 2) {
                keysToRemove.add(entry.getKey());
            }
        }

        // 统一删除
        for (String key : keysToRemove) {
            map.remove(key);
        }

        System.out.println(map); // 输出: {A=1, C=3}
    }
}

说明：

需要额外的集合（如 HashSet）来存储需要删除的键。
不推荐，因为代码冗长且效率较低。

四、使用 Stream（Java 8+）

如果你喜欢函数式编程风格，可以使用 Stream 来过滤并生成新的 HashMap。

import java.util.HashMap;
import java.util.Map;
import java.util.stream.Collectors;

public class Main {
    public static void main(String[] args) {
        HashMap<String, Integer> map = new HashMap<>();
        map.put("A", 1);
        map.put("B", 2);
        map.put("C", 3);

        // 使用 Stream 过滤并生成新的 HashMap
        Map<String, Integer> filteredMap = map.entrySet()
                .stream()
                .filter(entry -> entry.getValue() != 2)
                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

        System.out.println(filteredMap); // 输出: {A=1, C=3}
    }
}