Подтвердить что ты не робот

Двойной поток

Я хочу удвоить Stream (no DoubleStream). Смысл: я начинаю с потока и хочу получить новый поток, где каждый элемент старого потока передается дважды. Так 1,2,3,4,4,5 дает нам 1,1,2,2,3,3,4,4,4,4,5,5. Есть ли такая операция потока?

4b9b3361

Ответ 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 более непосредственно ответили на вопрос.