时间戳

首页 > 解决方案 > java 10中的Lazy flatMap实现算法

java - java 10中的Lazy flatMap实现算法

问题描述

我知道 java 流，并尝试实现 map、filter、fold（以自定义函数作为参数）、严格和惰性评估方式。

但是我无法在 java 中实现 flatmap 的惰性实现。法线贴图、过滤器、折叠只是在主迭代器（如果它的列表）上运行的组合函数，如果传入值为空，则丢弃函数的应用。

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

java 10 中的lazy flatMap 是如何实现的？有没有关于算法的文档？

谢谢。

标签： javafunctional-programming

解决方案

如果要实现lazy flatMap，最重要的部分是提供正确的实现Iterator。此实现可能如下所示：

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

因此，下一个元素的检索被推迟到调用hasNextor的那一刻。next

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

推荐阅读