Java之Map的keySet()方法避坑

  • Post author:
  • Post category:java

说明:在使用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());

    }

运行结果:
在这里插入图片描述


版权声明:本文为qq_42991839原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。