Java之Map的keySet()方法避坑

最新推荐文章于 2024-07-13 15:29:04 发布

爱北的琳儿

最新推荐文章于 2024-07-13 15:29:04 发布

阅读量1.5k

点赞数 1

文章标签： java 开发语言

本文链接：https://blog.csdn.net/qq_42991839/article/details/126118529

版权

说明：在使用map的keySet()方法，获得set集合并遍历，在循环中使用remove操作map集合导致报错
在这里插入图片描述

错误代码展示：

public static void main(String[] args) {


        // 新建一个map，分别存5个值，以Integer类型作为key，value随便
        Map<Integer, String> map = new HashMap<>();
        for (int i = 0; i < 5; i++) {
            map.put(i + 1, "aaa");
        }
        // 获取到map所有的key，并存入set集合中
        Set<Integer> set = map.keySet();
        // 遍历set
        for (Integer integer : set) {
            // 满足integer大于2的就从map集合中删除
            if (integer > 2) {
                map.remove(integer);
            }
            // 打印当前key
            System.out.println(integer);
        }

    }

}

原因：map.keySet()使用的是引用传递相当于map中的key在内存中的地址只和set集合中元素在内存中的地址值是同一个。因此我们移除map中的key相当于是移除了set集合中的元素。所以导致set集合的值消失和set的size变小无法继续执行循环中的工作。
解决：只需要将set集合的值循环出来重新存入新的Set集合（不重复）或者List集合（可以重复），根据需求选择集合。我这里选择的是Set集合。

解决方案代码：

public static void main(String[] args) {


        // 新建一个map，分别存5个值，以Integer类型作为key，value随便
        Map<Integer, String> map = new HashMap<>();
        for (int i = 0; i < 5; i++) {
            map.put(i + 1, "aaa");
        }
        // 获取到map所有的key，并存入set集合中
        Set<Integer> set = map.keySet();

        // 存入新的集合中
        Set<Integer> newSet = new HashSet<>();
        for (Integer integer : set) {
            newSet.add(integer);
        }

        // 遍历set
        for (Integer integer : newSet) {
            // 满足integer大于2的就从map集合中删除
            if (integer > 2) {
                map.remove(integer);
            }
            // 打印当前key
            System.out.println(integer);
        }
        System.err.println("set的size：" + set.size());

    }