Java TreeMap获得第K个最小键

Question

我正在使用TreeMap，并希望获取第K个最小键。 我编写了以下实现：

public static Long kthKey(TreeMap<Long, MyObject> treeMap, int kth) {
    Long currentKey = treeMap.firstKey();
    for (int i = 1; i < kth; i++) {
        currentKey = treeMap.higherKey(currentKey);
    }
    return currentKey;
}

就内存而言，这是非常有效的，但就运行时间而言可能并非如此。

有没有更有效的方法可以做到这一点？

Answer 1

如果允许使用迭代器，则可以执行此操作。

public <T> static Long kthKey(Map<T, ?> map, int kth) {
    for(T key: map.keySet())
       if(kth-- < 1) return key;
    return null;
}

一个迭代器的成本不是很高，如果您需要避免这种情况，我建议您看一下如何调用此方法并对其进行优化，例如说您拥有

for (int k = 0; k < mapsize(); k++) {
    Long l = kthKey(map, k);

您完全不需要调用此方法。