Scala Iterator/Stream с известным размером

Question

У меня есть Scala Iterator (я могу использовать любой тип данных для своей цели), который я нарисовал из Seq. Я хотел бы сохранить его, пока не буду готов, но я также хотел бы получить его размер в O (1). Мне было интересно, есть ли встроенный способ настройки, который позволил бы мне сохранить размер итератора на истребимом объекте.

Answer 1

Ваш лучший выбор - обернуть его в другой итератор, если вы можете оплатить стоимость дополнительной косвенности.

class SizedIterator[A](underlying: Iterator[A], val initalSize: Int) extends Iterator[A] { 
    def next = underlying.next 
    def hasNext = underlying.hasNext 
} 

, а затем

new SizedIterator(mySeq.iterator, mySeq.length) 

Имейте в виду, однако, что если вы карту или что-то ваш новый SizedIterator вы будете в конечном итоге с простым Iterator и уже не знаю, как долго initialSize было.

Также имейте в виду, что вы не знаете, сколько итератора потребляется, поэтому initialSize является верхней границей размера, но когда вы его используете, может не быть ни одного.

В качестве альтернативы, вы можете

mySeq.iterator.zipWithIndex.map{ case (x,i) => (x, mySeq.length-1) }.take(mySeq.length) 

производить Iterator, которое пары элемента и число элементов, оставшихся (в том числе, что один - вы будете никогда не ударил нулевой таким образом).

Если вы просто хотите что-то с известным размером, который расскажет вам в O(1), в .length метод на Seq приведение к Iterable рассосется так быстро, как метод будет иметь, когда он был Seq --this весь смысл от переопределяющих методов - и поэтому не будет (обычно) потреблять новый Iterator каждый раз. Но это требует от вас держать Iterable, а не только Iterator.