Я хочу удвоить Stream (no DoubleStream). Смысл: я начинаю с потока и хочу получить новый поток, где каждый элемент старого потока передается дважды. Так 1,2,3,4,4,5 дает нам 1,1,2,2,3,3,4,4,4,4,5,5. Есть ли такая операция потока?
Двойной поток
Ответ 1
Создайте внутренний поток, который будет содержать текущий элемент два раза, и flatMap этот поток.
stream.flatMap(e -> Stream.of(e,e))
Если вы хотите умножить количество элементов на n
, вы можете создать такой служебный метод, как этот:
public static <T> Stream<T> multiplyElements(Stream<T> in, int n) {
return in.flatMap(e -> IntStream.range(0, n).mapToObj(i -> e));
// we can also use IntStream.rangeClosed(1, n)
// but I am used to iterating from 0 to n (excluding n)
}
(но попытайтесь использовать лучшее имя для этого метода, так как текущий может быть неоднозначным)
Пример использования:
multiplyElements(Stream.of(1,2), 3).forEach(System.out::println);
Вывод:
1
1
1
2
2
2
Ответ 2
Вы можете создать поток из 2 элементов для каждого исходного элемента и flatMap it:
List<Integer> list = Arrays.asList(1, 2, 3, 4, 4, 5);
List<Integer> doubled = list.stream().flatMap(i -> Stream.of(i, i)).collect(toList());
Ответ 3
Вот простой пример того, что biziclop описал в комментариях.
static <E> Collection<E> multiply(Collection<E> source, int count) {
return new AbstractCollection<E>() {
@Override
public int size() {
return count * source.size();
}
@Override
public Iterator<E> iterator() {
return new Iterator<E>() {
final Iterator<E> it = source.iterator();
E next;
int i = 0;
@Override
public boolean hasNext() {
return i < size();
}
@Override
public E next() {
if (hasNext()) {
if ((i % count) == 0) {
next = it.next();
}
++i;
return next;
} else {
throw new NoSuchElementException();
}
}
};
}
};
}
(Рабочий пример на Ideone.)
CW'd, так как это была не моя идея, а предложения flatMap
более непосредственно ответили на вопрос.