我试图在List<T>
中找到一个尾函数,但我找不到。我最终做到了。
fun <T> List<T>.tail() = this.takeLast(this.size -1)
有没有更好的方法可以做到这一点?
Kotlin 没有内置的 List<T>.tail()
函数,因此实现自己的扩展函数是唯一的方法。虽然你的实现很好,但可以简化一点:
fun <T> List<T>.tail() = drop(1)
或者,您可以定义扩展属性,而不是扩展函数:
val <T> List<T>.tail: List<T>
get() = drop(1)
val <T> List<T>.head: T
get() = first()
然后像这样使用它:
val list = listOf("1", "2", "3")
val head = list.head
val tail = list.tail
你和@Vladimir Mironov 的解决方案将起作用,但它们会自动创建原始列表的预先副本(没有第一个元素),这对于较大的列表可能需要很长时间。我会用一个包装器List
类来定义它,该类将其方法委托给包装的方法,使用索引调整忽略第一个元素:
private class TailList<T> (private val list: List<T>) : List<T> {
override val size: Int
get() = list.size -1
override fun isEmpty(): Boolean = size == 0
override fun iterator(): Iterator<T> = listIterator()
override fun listIterator(): ListIterator<T> = list.listIterator(1)
override fun listIterator(index: Int): ListIterator<T> = list.listIterator(index + 1)
override fun subList(fromIndex: Int, toIndex: Int): List<T> = list.subList(fromIndex + 1, toIndex + 1)
override fun lastIndexOf(element: T): Int = list.lastIndexOf(element) - 1
override operator fun get(index: Int): T = list[index + 1]
// The following member functions require the copy of a new list
override fun containsAll(elements: Collection<T>): Boolean = tailList.containsAll(elements)
override fun contains(element: T): Boolean = tailList.contains(element)
override fun indexOf(element: T): Int = tailList.indexOf(element)
private val tailList by lazy { ArrayList(this) } // makes a proper copy the elements this list represents
}
您可能会注意到注释后部分中的功能最终仍然会生成一个急切的副本。我这样做只是为了简单起见。为了记忆,我做了一个lazy
tailList
属性
它们都可以通过手动迭代集合来实现,而不是执行某种委托。如果这是你喜欢的,我相信你可以弄清楚。
这样,头部和尾部属性变为:
val <T> List<T>.tail: List<T>
get() =
if(this.isEmpty())
throw IllegalStateException("Cannot get the tail of an empty List")
else
TailList(this)
val <T> List<T>.head: T
get() = this[0] // or first()
如果你真的需要它,我可以添加一个更新来制作最后三个成员函数,这样他们就不会急于复制。
编辑:注意:如果你遵循了 Kotlin 到目前为止遵循的惯例,你就不会像这样让List
的尾巴变得懒惰,因为它们的所有功能List
都会做出渴望的副本。相反,特别是如果您使用 head
和 tail
递归迭代列表,我会看看您是否可以以某种方式尝试Sequence
包装器的想法。 Sequence
存在的全部意义在于懒惰的收藏工作。
编辑2:显然 sublist() 创建了一个视图,因此已经很懒了。从本质上讲,我刚刚教了您如何创建子列表的实现,只是我将其缩小到尾部。
因此,在这种情况下,只需对尾部函数使用 sublist()。
如果使用不可变列表,则完全安全,并且占用内存更少,只需使用:
fun <T> List<T>.tail(): List<T> =
if (isEmpty()) throw IllegalArgumentException("tail called on empty list")
else subList(1, count())
如果你需要头部和尾部(这是一个基本的序列操作),并且不急切地使用流,你可能想要一个序列。
fun <T> Sequence<T>.headTail() =
HeadTailSequence(this)
.let { it.head to it.iterator.asSequence() }
private class HeadTailSequence<T>(parent: Sequence<T>) : Sequence<T> {
val iterator = parent.iterator()
val head: T? = iterator.next()
override fun iterator(): Iterator<T> {
return object : Iterator<T> {
override fun hasNext(): Boolean = iterator.hasNext()
override fun next(): T = if (iterator.hasNext()) throw NoSuchElementException() else iterator.next()
}
}
}
可迭代的扩展可以用类似的方式完成。