问题描述
假设我有一个地图:Map<Int,String>
,我知道这个地图是如何创建的,并且它保留了迭代顺序。如何迭代给定范围内的所有元素?
val map: Map<Int,String> = mapOf(
1 to "one",2 to "two",3 to "three",4 to "four")
// Let's say I want to get all values after key 3.
val iterator = map.iterator()
while (iterator.next().key != 3) {}
val itemsAfter3 = iterator.asSequence().map { it.value }
除了丑之外,就是O(n)。有没有更好、更优雅、更高效的方式?
我正在考虑用 map.keys.indexOf()
做点什么,但这里的复杂性似乎也是线性的。
解决方法
永远不要依赖内部实现,将来可能会更改。如果您需要在迭代期间保留插入顺序的地图,请明确声明:
val map: Map<Int,String> = linkedMapOf(
1 to "one",2 to "two",3 to "three",4 to "four"
)
获取特定索引后的所有值很容易:
val valuesAfterIndex3 = map.values.drop(3)