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

Question

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

我可以进行迭代，但是必须检查null等很繁琐。

Answer 1

为什么要检查空值？

如果您使用Google Collections Library ，则可以使用Iterables.limit 。

编辑：由于某些原因，GCL页面不包含limit ，但是它位于Guava中 （已有效替换了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，boolean，K，boolean）