浅谈java中遍历Map的几种方法

青旅半醒 2022-04-13 06:17 250阅读 0赞

java中的map遍历有多种方法，从最早的Iterator，到java5支持的foreach,再到java8 Lambda，让我们一起来看下具体的用法以及各自的优缺点

先初始化一个map

public class TestMap {
      public static Map<Integer, Integer> map = new HashMap<Integer, Integer>();
    }

## keySet values ##

如果只需要map的key或者value，用map的keySet或values方法无疑是最方便的

// KeySet 获取key
      public void testKeySet() {
        for (Integer key : map.keySet()) {
          System.out.println(key);
        }
      }
      // values 获取value
      public void testValues() {
        for (Integer value : map.values()) {
          System.out.println(value);
        }
      }

## keySet get(key) ##

如果需要同时获取key和value，可以先获取key,然后再通过map的get(key)获取value

需要说明的是，该方法不是最优选择，一般不推荐使用

// keySet get(key) 获取key and value
      public void testKeySetAndGetKey() {
        for (Integer key : map.keySet()) {
          System.out.println(key + ":" + map.get(key));
        }
      }

## entrySet ##

通过对map entrySet的遍历，也可以同时拿到key和value，一般情况下，性能上要优于上一种,这一种也是最常用的遍历方法

// entrySet 获取key and value
      public void testEntry() {
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
          System.out.println(entry.getKey() + ":" + entry.getValue());
        }
      }

## Iterator ##

对于上面的几种foreach都可以用Iterator代替，其实foreach在java5中才被支持，foreach的写法看起来更简洁

但Iterator也有其优势：在用foreach遍历map时，如果改变其大小，会报错，但如果只是删除元素，可以使用Iterator的remove方法删除元素

// Iterator entrySet 获取key and value
      public void testIterator() {
        Iterator<Map.Entry<Integer, Integer>> it = map.entrySet().iterator();
        while (it.hasNext()) {
          Map.Entry<Integer, Integer> entry = it.next();
          System.out.println(entry.getKey() + ":" + entry.getValue());
          // it.remove(); 删除元素
        }
      }

## Lambda ##

java8提供了Lambda表达式支持，语法看起来更简洁，可以同时拿到key和value，不过，经测试，性能低于entrySet,所以更推荐用entrySet的方式

// Lambda 获取key and value
      public void testLambda() {
        map.forEach((key, value) -> {
          System.out.println(key + ":" + value);
        });
      }

## 简单性能测试 ##

用10万条数据，做了一个简单性能测试，数据类型为Integer，map实现选取HashMap

static {
        for (int i = 0; i < 100000; i++) {
          map.put(i, 1);
        }
      }

测试结果如下：

KeySet：           392
    Values：           320
    keySet get(key)：  552
    entrySet：         465
    entrySet Iterator：508
    Lambda：           536

需要说明的是，map存储的数据类型，map的大小，以及map的不同实现方式都会影响遍历的性能，所以该测试结果仅供参考

## 总结 ##

如果只是获取key，或者value，推荐使用keySet或者values方式

如果同时需要key和value推荐使用entrySet

如果需要在遍历过程中删除元素推荐使用Iterator

如果需要在遍历过程中增加元素，可以新建一个临时map存放新增的元素，等遍历完毕，再把临时map放到原来的map中