Double a stream

Question

Tôi muốn tăng gấp đôi số Stream (không DoubleStream). Có nghĩa là tôi bắt đầu bằng luồng và muốn nhận luồng mới trong đó mỗi phần tử của luồng cũ được phát trực tiếp hai lần. Vì vậy, 1,2,3,4,4,5 cho chúng ta 1,1,2,2,3,3,4,4,4,4,5,5. Có hoạt động luồng như vậy không?

Answer 1

Tạo luồng bên trong chứa phần tử hiện tại hai lần và flatMap luồng này.

stream.flatMap(e -> Stream.of(e,e)) 

Nếu bạn muốn để nhân số phần tử bằng n bạn có thể tạo ra một phương pháp hữu ích như thế này:

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) 
} 

(nhưng cố gắng sử dụng một tên tốt hơn cho phương pháp này, vì hiện tại người ta có thể được mơ hồ)

Cách sử dụng Ví dụ:

multiplyElements(Stream.of(1,2), 3).forEach(System.out::println); 

Output:

1 
1 
1 
2 
2 
2 

Answer 2

Bạn có thể tạo một dòng của 2 yếu tố cho mỗi phần tử gốc và flatMap nó:

List<Integer> list = Arrays.asList(1, 2, 3, 4, 4, 5); 
List<Integer> doubled = list.stream().flatMap(i -> Stream.of(i, i)).collect(toList()); 

Answer 3

Dưới đây là một ví dụ đơn giản về những gì biziclop đã được mô tả trong các ý kiến.

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(); 
        } 
       } 
      }; 
     } 
    }; 
} 

(ví dụ làm việc trên Ideone.)

CW'd vì nó không phải là ý tưởng của tôi và flatMap gợi ý trực tiếp hơn trả lời câu hỏi.