时间戳

首页 > 解决方案 > 如何创建一个集合和迭代器，它会遍历某些值直到达到一个值？

java - 如何创建一个集合和迭代器，它会遍历某些值直到达到一个值？

问题描述

我有一个TreeMap<Token, ArrayList<Token>>并且我想遍历地图直到满足要求的特定键。我知道以下内容可用于获取地图的值：

Collection c = bigrams.values();
Iterator itr = c.iterator();

while (itr.hasNext()){
    System.out.println(itr.next());

但是，我希望能够使用链接到迭代器的键来遍历映射，并根据其对键检查每个值。由于bigrams.values()检索二元组的每个元素的值，我该如何更改它以检索键而不是值？

标签： javaiteratortreemap

解决方案

您应该选择使用Map.keySet()or Map.entrySet()：

Map.keySet()返回一个包含地图所有键的集合。然后，您可以使用Map.get()来获取给定键的值：

for (Token key: bigrams.keySet()) {
    List<Token> list = bigrams.get(key);
    System.out.println(key + ": " + list);
}

Map.entrySet()在您的地图中返回一组所有对，因此无需Map.get()与此一起使用：

for (Map.Entry<Token, List<Token>> entry : bigrams.entrySet()) {
    System.out.println(entry.getKey() + ": " + entry.getValue());
}

最后，您还可以为此使用 Java Stream API。您还可以使用它非常轻松地过滤内容。例如，要在其值列表中查找包含给定标记的所有标记：

bigrams.entrySet().stream()
        .filter(e -> e.getValue().contains(tokenToFind))
        .forEach(e -> System.out.println(e.getKey() + ": " + e.getValue()));

推荐阅读