java - 采访 : Design an iterator for a collection of collections

Question

为java中的集合设计一个迭代器。迭代器应该隐藏嵌套，允许您迭代属于所有集合的所有元素，就好像您在处理单个集合一样

Answer 1

这是一个老问题，但如今(2019 年)我们有了 JDK8+ 好东西。特别是，我们有流，这使得这个任务变得简单:

public static <T> Iterator<T> flatIterator(Collection<Collection<T>> collections) {

    return collections.stream()
            .filter(Objects::nonNull)
            .flatMap(Collection::stream)
            .iterator();
}

我正在过滤 null 内部集合，以防万一...

---

编辑: 如果您还想从内部集合中过滤出 null 元素，只需在 flatMap 之后添加一个额外的非空过滤器:

return collections.stream()
        .filter(Objects::nonNull)
        .flatMap(Collection::stream)
        .filter(Objects::nonNull)
        .iterator();