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

Posted 2020-12-07 骇客黑界

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

先初始化一个map

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

keySet values

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

 1 // KeySet 获取key
 2   public void testKeySet() {
 3     for (Integer key : map.keySet()) {
 4       System.out.println(key);
 5     }
 6   }
 7   // values 获取value
 8   public void testValues() {
 9     for (Integer value : map.values()) {
10       System.out.println(value);
11     }
12   }

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

测试结果如下：

1 KeySet：           392
2 Values：           320
3 keySet get(key)：  552
4 entrySet：         465
5 entrySet Iterator：508
6 Lambda：           536

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

总结

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

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

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

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