10

イテレータを拡張して新しいメソッドを作成しようとしています。このメソッドは、最後の要素takeWhileInclusiveと同じように動作しますが、含まれています。takeWhile

私の問題は、イテレータを拡張して、遅延評価したい新しいイテレータを返すためのベストプラクティスです。私は通常C＃のバックグラウンドから来IEnumerableてキーワードを使用してyieldいますが、そのようなオプションはScalaには存在しないようです。

たとえば、私は持つことができます

List(0,1,2,3,4,5,6,7).iterator.map(complex time consuming algorithm).takeWhileInclusive(_ < 6)

したがって、この場合、takeWhileInclusive6より大きい結果が得られるまで、値の述語のみが解決され、この最初の結果が含まれます。

これまでのところ：

object ImplicitIterator {
  implicit def extendIterator(i : Iterator[Any]) = new IteratorExtension(i)
}

class IteratorExtension[T <: Any](i : Iterator[T]) {
  def takeWhileInclusive(predicate:(T) => Boolean) = ?
}
4

5 に答える 5

11

spanの方法を使用して、Iteratorこれをかなりきれいに行うことができます。

class IteratorExtension[A](i : Iterator[A]) {
  def takeWhileInclusive(p: A => Boolean) = {
    val (a, b) = i.span(p)
    a ++ (if (b.hasNext) Some(b.next) else None)
  }
}

object ImplicitIterator {
  implicit def extendIterator[A](i : Iterator[A]) = new IteratorExtension(i)
}

import ImplicitIterator._

たとえば、を(0 until 10).toIterator.takeWhileInclusive(_ < 4).toList与えます。List(0, 1, 2, 3, 4)

于 2012-02-17T14:44:48.673 に答える
7

これは、可変ソリューションが優れていると思う1つのケースです。

class InclusiveIterator[A](ia: Iterator[A]) {
  def takeWhileInclusive(p: A => Boolean) = {
    var done = false
    val p2 = (a: A) => !done && { if (!p(a)) done=true; true }
    ia.takeWhile(p2)
  }
}
implicit def iterator_can_include[A](ia: Iterator[A]) = new InclusiveIterator(ia)
于 2012-02-17T15:31:18.873 に答える
3

fold以下では、タプルに乗るためにscalazが必要です(A, B)

scala> implicit def Iterator_Is_TWI[A](itr: Iterator[A]) = new { 
     | def takeWhileIncl(p: A => Boolean) 
     |   = itr span p fold (_ ++ _.toStream.headOption)
     | }
Iterator_Is_TWI: [A](itr: Iterator[A])java.lang.Object{def takeWhileIncl(p: A => Boolean): Iterator[A]}

ここでそれは働いています：

scala> List(1, 2, 3, 4, 5).iterator takeWhileIncl (_ < 4)
res0: Iterator[Int] = non-empty iterator

scala> res0.toList
res1: List[Int] = List(1, 2, 3, 4)

あなたはこのようにペアの上にあなた自身の折り目を転がすことができます：

scala> implicit def Pair_Is_Foldable[A, B](pair: (A, B)) = new { 
    |    def fold[C](f: (A, B) => C): C = f.tupled(pair) 
    |  } 
Pair_Is_Foldable: [A, B](pair: (A, B))java.lang.Object{def fold[C](f: (A, B) => C): C}
于 2012-02-17T20:34:40.447 に答える
2 
class IteratorExtension[T](i : Iterator[T]) {
  def takeWhileInclusive(predicate:(T) => Boolean) = new Iterator[T] {
    val it = i
    var isLastRead = false

    def hasNext = it.hasNext && !isLastRead
    def next = {
      val res = it.next
      isLastRead = !predicate(res)
      res
    }
  }
}

そして、あなたの暗黙の中にエラーがあります。ここで修正されました：

object ImplicitIterator {
  implicit def extendIterator[T](i : Iterator[T]) = new IteratorExtension(i)
}
于 2012-02-17T15:25:37.390 に答える
0 
scala> List(0,1,2,3,4,5,6,7).toStream.filter (_ < 6).take(2)
res8: scala.collection.immutable.Stream[Int] = Stream(0, ?)

scala> res8.toList 
res9: List[Int] = List(0, 1)

更新後：

scala> def timeConsumeDummy (n: Int): Int = {
     | println ("Time flies like an arrow ...") 
     | n }
timeConsumeDummy: (n: Int)Int

scala> List(0,1,2,3,4,5,6,7).toStream.filter (x => timeConsumeDummy (x) < 6) 
Time flies like an arrow ...
res14: scala.collection.immutable.Stream[Int] = Stream(0, ?)

scala> res14.take (4).toList 
Time flies like an arrow ...
Time flies like an arrow ...
Time flies like an arrow ...
res15: List[Int] = List(0, 1, 2, 3)

timeConsumeDummyは4回呼び出されます。私は何かが足りないのですか？

于 2012-02-17T14:37:35.400 に答える