java 10中的懒惰平面图实现算法

Question

我了解java streams，并尝试实现map、filter、fold（使用自定义函数作为参数），包括严格的和惰性的计算方法。

然而，我无法在java中实现flatmap的惰性实现。Normal map、filter、fold只是在主迭代器（如果是它的列表）上运行的组合函数，如果传入值为null，则丢弃函数的apply。

但是flatMap输入函数会生成另一个列表（流），需要将其展平，

lazy flatMap是如何在Java10中实现的？有关于算法的文件吗？

谢谢

Answer 1

如果要实现lazyflatMap，最重要的部分是提供正确的迭代器实现。此实现可以如下所示：

final class FlatMappedIterator<A, B> implements Iterator<B> {
    private final Iterator<A> iterator;
    private final Function<A, Iterable<B>> f;
    private Iterator<B> targetIterator; // Iterator after applying `f` to element of type A

    FlatMappedIterator(Iterator<A> iterator, Function<A, Iterable<B>> f) {
        this.iterator = iterator;
        this.f = f;
    }

    @Override
    public boolean hasNext() {
        if (targetIterator != null && targetIterator.hasNext()) {
            return true;
        } else if (iterator.hasNext()) {
            A next = iterator.next();
            Iterable<B> targetIterable = f.apply(next);
            targetIterator = targetIterable.iterator();
            return targetIterator.hasNext();
        } else {
            return false;
        }
    }

    @Override
    public B next() {
        if (hasNext()) {
            return targetIterator.next();
        } else {
            throw new NoSuchElementException();
        }
    }
}

因此，下一个元素的检索被推迟到调用hasNext或Next的时刻。

然后需要实现flatMap函数本身。但这很容易。我把它作为练习留给读者：）