获取 TreeMap 的键集子集的简单方法（Java 中）

Question

是否可以从 TreeMap 中提取前 n 个（我使用 top 因为我相信 TreeMap 已排序）关键元素，而不使用 Iterator 对象进行迭代。

我可以进行迭代，但是必须检查空值等很乏味。

Answer 1

为什么要检查空值？

如果您使用 Google 收藏库，则可以使用 Iterables.limit 。

编辑：由于某种原因，GCL 页面不包含 limit，但它位于 Guava< /a> （实际上已经取代了 GCL） - 它仍然是 Iterables.limit。

Answer 2

您可以使用 .subMap(low,high).keySet() 或 .headMap(high).keySet() 轻松获取从一个键到另一个键的键子集。

找出第 n 个正确的高调更加困难，因为没有直接的方法，而迭代是唯一可靠的方法。

（此代码未经测试）

public <K,V> SortedMap<K,V> subMap(SortedMap<K,V> map, int n) {
  Iterator<K> it = map.keySet().iterator();
  for (int i = 0; i<n && it.hasNext(); i++) {
    it.next();
  }
  if (it.hasNext()) {
    return map.headMap(it.next());
  } else {
    return map
  }
}

Answer 3

您可以使用：

[subMap 方法][1]

public NavigableMap<K,V> subMap(K fromKey,boolean fromInclusive,K toKey,boolean toInclusive)

[1]: http://java.sun.com/javase/6/docs/api/java/util/TreeMap.html#subMap(K，布尔值，K，布尔值)