java - 使用流递归展平列表

Question

我有内部节点和终端节点的树状结构:

public interface Node
{
}

public class InternalNode implements Node {
    private List<Node> nodes;
}

public class TerminalNode implements Node {
    private String label;
}

我现在有一个 List<Node>我想压平。在这里，扁平化意味着我想用它的 child 递归地替换一个内部节点，直到所有内部节点都被终端替换。

我想出了这个函数:

private static List<Node> flatten(final List<Node> nodes) {
    return nodes
            .stream()
            .map(node -> {
                if (node instanceof InternalNode) {
                    return flatten(((InternalNode) node).getNodes());
                }
                return Collections.singletonList(node);
            })
            .flatMap(List::stream)
            .collect(Collectors.toList());
}

这似乎完成了它的工作。但是，我想知道是否有更好的实现方式。我首先必须包装一个 TerminalNode 似乎很奇怪通过 List<TerminalNode> 进入单例列表(类型为 Collections.singletonList(node) )然后我必须通过 flatMap(List::stream) 再次将该单例列表转换回节点.

有没有办法避免这种无用的Collections.singletonList(node)其次是 flatMap(List::stream)对于终端节点？

Answer 1

你可以直接使用 flatMap:

private static Stream<TerminalNode> flatten(final List<Node> nodes) {
    return nodes
            .stream()
            .flatMap(node -> {
                if (node instanceof InternalNode) {
                    return flatten(((InternalNode) node).getNodes());
                }
                return Stream.of((TerminalNode) node);
            });
}

如果你想要一个列表，你可以只收集那个方法调用的结果。