用流递归地展平列表

Question

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

public interface Node
{
}

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

public class TerminalNode implements Node {
    private String label;
}

我现在有一个列表

我想出了这个函数：

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

有没有办法避免这个无用的Collections.singletonList（节点）后面跟着平映射（列表::流）的终端节点？




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

如果想要列表，只需收集该方法调用的结果即可。