在Scala中編寫帶有偏移量的通用takeWhile

Question

我有一個takeWhile的用例，但是我希望在謂詞爲true後保留固定數量的項目。我也試圖把它寫成儘可能通用的類型的集合。因此，像：

def takeWhileWithOffset[A, Iter[_] <: Iterable[A]](iter: Iter[A], p: A => Boolean, offset: Int) 

我選擇Iterable爲界，是因爲希望它與Stream工作。但是我很難搞清楚如何做這個工作。如果我使用的是嚴格收集，如果偏移量不是正數，我可以使用dropRight。但是Iterable沒有dropRight。

正面的情況更棘手。我可以使用sliding來有效抓取未來物品，然後在takeWhile終止後使用init和lastOption。但是Iterable沒有init和lastOption。

那麼，什麼是棘手的是，我希望我的方法偷懶，但採取的是迭代的項目可以作爲一個嚴格的收集要處理的事實的優點 - 但只有的takeWhile終止。有沒有辦法做到這一點？

Answer 1

span和take：

scala> val it = (1 to 100).iterator 
it: Iterator[Int] = non-empty iterator        

scala> val (a, b) = it.span(_ < 10) 
a: Iterator[Int] = non-empty iterator 
b: Iterator[Int] = unknown-if-empty iterator 

scala> val res = a ++ b.take(5) 
res: Iterator[Int] = non-empty iterator 

scala> res.toList 
res0: List[Int] = List(1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14)