基于两个有序元素在两个其他流中是否相等来创建流

时间:2015-05-15 17:31:22

标签: java java-8 java-stream

我正在尝试学习如何在我通常会强制执行的情况下使用Java Streams。

所以,如果我的流1看起来像这样:

1 4 3 5 6 9 4 1 ...

和流2看起来像这样

1 0 3 0 0 9 4 0 ...

我希望它创建一个只包含相同元素的第三个流,所以:

1 3 9 4 ...

3 个答案:

答案 0 :(得分:3)

除了将它们转换为迭代器并以这种方式手动执行之外,对于流来说这看起来并不可行。实际上没有办法将两个流组合起来。

答案 1 :(得分:3)

如果您的来源是随机访问列表,则可以执行以下操作:

List<Integer> list1 = Arrays.asList(1, 4, 3, 5, 6, 9, 4, 1, ...);
List<Integer> list2 = Arrays.asList(1, 0, 3, 0, 0, 9, 4, 0, ...);

IntStream.range(0, list1.size())
    .mapToObj(i -> list1.get(i).equals(list2.get(i)) ? list1.get(i) : null)
    .filter(Objects::nonNull).collect(Collectors.toList());

在我的StreamEx库中,有一个专门用于此案例的快捷方式:

StreamEx.zip(list1, list2, (a, b) -> a.equals(b) ? a : null).nonNull().toList();

此外,如果您不关心并行处理,可以使用jOOL库,它允许您将两个流压缩在一起:

Seq.of(list1).zip(Seq.of(list2), (a, b) -> a.equals(b) ? a : null)
             .filter(Objects::nonNull).toList();

这对并行流表现不佳,但适用于任何流源(不仅是随机访问列表或数组)。

答案 2 :(得分:1)

你必须压缩你的流 - 我可以推荐你protonpack小库。这是一种测试方法:

import com.codepoetics.protonpack.StreamUtils;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.*;
import org.junit.Test; 

// ... skipped test class declaration
@Test
public void shouldRetrieveOnlyEqual() {
    // given
    Stream<Integer> s1 = Stream.of(1, 4, 3, 5, 6, 9, 4, 1);
    Stream<Integer> s2 = Stream.of(1, 0, 3, 0, 0, 9, 4, 0);

    // when
    List<Integer> actual = StreamUtils.zip(s1, s2, (v1, v2) -> v1.equals(v2) ? v1 : null)
            .filter(Objects::nonNull).collect(Collectors.toList());

    // then
    assertThat(actual, contains(1, 3, 9, 4));
}