如何使用Java 8 Streams以特定的方式实现zip？

Question

我试图弄清楚如何使用Streams API实现一个zip函数，该函数以无限多的int[]作为参数；从每个元素中获取i'th元素；将它们放在一个元组中（显然需要一个自定义的元组对象-我有），并返回一个元组列表（即list）。

本质上，为：

{ 1, 2, 3 }
{ 4, 5, 6 }

建议的方法应该以java的形式返回：[元组（1，4），元组（2，5），元组（3，6）]。util。列表

这是一个函数，它以“正常”的方式完成了我试图做的事情：

/**
 * Return a list of tuples, where each tuple contains the i-th element
 * from each of the argument sequences.  The returned list is
 * truncated in length to the length of the shortest argument sequence.
 *
 * @param args  the array of ints to be wrapped in {@link Tuple}s
 * @return a list of tuples
 */
public static List<Tuple> zip(int[]... args) {
    List<Tuple> retVal = new ArrayList<>();
    // Find the array with the minimum size
    int minLength = Arrays.stream(args).map(i -> new Integer(i.length)).min((a, b) -> a.compareTo(b)).get();

    for(int i = 0;i < minLength;i++) {
        Tuple.Builder builder = Tuple.builder();
        for(int[] ia : args) {
            builder.add(ia[i]);
        }
        retVal.add(builder.build());
    }

    return retVal;
}

Answer 1

一种解决方案是在索引上创建一个Stream，并使用mapToObj将每个int映射到Tuple中。此外，由于您已经有了Builder对象，我们可以利用它将元素收集到其中。

假设我们添加一个方法<代码>元组。建设者addAll（Tuple.Builder other）其目的是将一个生成器添加到另一个生成器，我们可以有以下代码：

public static List<Tuple> zip(int[]... args) {
    // Find the array with the minimum size
    int minLength = Arrays.stream(args).mapToInt(i -> i.length).min().orElse(0);

    return IntStream.range(0, minLength)
                    .mapToObj(i -> 
                       Arrays.stream(args)
                             .mapToInt(ia -> ia[i])
                             .collect(Tuple::builder, Tuple.Builder::add, Tuple.Builder::addAll)
                             .build()
                    ).collect(Collectors.toList());
}

（如果您不想支持并行执行，您可以使用（b1， b2）-

作为补充说明，查找最小数组大小的代码可以简化：您不需要将其装箱为整数，只需将每个数组映射到其长度，并使用最小值（）。这将返回一个可选值；我没有获取它的值（如果流为空，则可能引发异常），而是使用了orElse（0），以便在流为空的情况下，返回一个空列表。