Java中使用parallelStream替换递归方法中的for循环

嘿，这个问题我刚好处理过，咱们一步步来搞定它！

首先，先回顾你原来的逻辑：遍历chList里的每个子节点，把每个子节点调用getLeavesBelow()返回的叶子列表全部合并到结果中。现在要换成并行流，核心问题就是怎么把多个子列表合并成一个大列表，替代原来的addAll操作。

这里有两种简洁且符合并行流规范的实现方式：

方案一：用flatMap扁平化流（推荐） #

这个方式最直观，直接把每个子节点返回的列表转换成流，再合并成一个统一的流，最后收集成ArrayList：

public ArrayList<SpecTreeNode> getLeavesBelow() {
    ArrayList<SpecTreeNode> result = new ArrayList<>();
    if (isLeaf()) {
        result.add(this);
    } else {
        // 并行处理子节点，扁平化所有叶子节点流后收集
        result.addAll(
            chList.parallelStream()
                  // 把每个子节点的叶子列表转成流，flatMap自动合并所有流
                  .flatMap(stn -> stn.getLeavesBelow().stream())
                  // 收集成ArrayList，和原代码的容器类型保持一致
                  .collect(Collectors.toCollection(ArrayList::new))
        );
    }
    return result;
}

为什么这么写？ #

• flatMap会把每个ArrayList<SpecTreeNode>拆解成单个SpecTreeNode的流，相当于把所有子节点的叶子节点“平铺”成一个流，完全替代了原来循环里的addAll逻辑。
• 并行流的线程安全由收集器自动处理，不用担心多线程操作容器的问题。

方案二：用collect的三参数重载（贴近原循环逻辑） #

如果你更想保留“先获取子列表，再合并”的思路，可以用collect的三个参数版本，直接模拟addAll的行为：

public ArrayList<SpecTreeNode> getLeavesBelow() {
    ArrayList<SpecTreeNode> result = new ArrayList<>();
    if (isLeaf()) {
        result.add(this);
    } else {
        result.addAll(
            chList.parallelStream()
                  .map(SpecTreeNode::getLeavesBelow)
                  // 三个参数分别是：容器创建、元素累加、容器合并
                  .collect(
                      ArrayList::new,       // 每个线程创建自己的临时ArrayList
                      ArrayList::addAll,    // 把每个子列表加到临时容器
                      ArrayList::addAll     // 并行时合并多个临时容器
                  )
        );
    }
    return result;
}

这个版本的细节： #

• 第一个参数ArrayList::new：并行流会为每个线程创建独立的临时容器，避免线程安全问题。
• 第二个参数ArrayList::addAll：把每个子节点返回的列表加到当前线程的临时容器里，和你原来的result.addAll(stn.getLeavesBelow())完全一致。
• 第三个参数ArrayList::addAll：当多个线程处理完各自的子任务后，把所有临时容器的内容合并成一个最终列表。

小提醒 #

并行流不是银子弹哦！如果chList的元素数量不多，并行处理的线程开销可能会超过它带来的收益。建议你根据实际的数据规模测试一下，再决定是否用并行流～

内容的提问来源于stack exchange，提问作者Steve W