如何递归flatMap流？[副本]

Question

我被要求检索树节点的每个叶节点。我很快想到我可以在一行中完成这项工作！

public Set<TreeNode<E>> getLeaves() {
    return getChildrenStream().flatMap(n -> n.getChildrenStream()).collect(toSet());
}

第一眼看上去很不错，但很快它就遇到了一个StackAfExcepetion，如果树深度达到~10，这是我无法接受的。后来我开发了一个没有递归和流的实现（但我的大脑被烤坏了），但我仍然想知道是否有一种方法可以用流做递归平面图，因为我发现不接触流内部是不可能做到的。它需要一个新的Op，比如RecursiveOps来做到这一点，否则我必须每一步都将所有结果收集到Set中，然后再对该Set进行操作：

Set<TreeNode<E>> prev = new HashSet<>();
prev.add(this);
while (!prev.isEmpty()) {
    prev = prev.stream().flatMap(n -> n.getChildrenStream()).collect(toSet());
}
return prev;

不像看起来那么好。流意味着管道。它的结果和中间结果直到添加了一个终端op才被计算。上述方法显然违反了这一原则。它也不像流那样容易并行化。我可以递归flatMap而不用手动计算所有中间结果吗？

PS1：TreeNode声明：

public class TreeNode<E> {
    // ...
    /**
     * Get a stream of children of the current node. 
     *
     */
    public Stream<TreeNode<E>> getChildrenStream(){
        // ...
    }

    public Set<TreeNode<E>> getLeaves() {
        // main concern
    }
}f

Answer 1

不完全确定这是否是您感兴趣的内容：

public static Set<TreeNode<String>> getAllLeaves(TreeNode<String> treeNode) {
    final Stream<TreeNode<String>> childrenStream = treeNode.getChildrenStream();
    if (childrenStream == null) {
        return new HashSet<>();
    }

    Set<TreeNode<String>> ownLeaves = treeNode.getLeaves();
    ownLeaves.addAll(childrenStream.flatMap(stringTreeNode -> getAllLeaves(stringTreeNode).parallelStream())
            .collect(Collectors.toSet()));

    return ownLeaves;
}

开箱即用，我看到了这个方法的一些不方便之处。它确实为最后一次迭代返回了一个空的Set，并且它正在创建流，就像它在平地图一样。然而，我相信这就是你正在寻找的，因为你正在考虑使用平地图，从那里你想得到一个递归创建的连接集，而在那里没有创建流。顺便说一句，我已经用-1000级别尝试过了，它仍然工作得相当快，没有问题。