如何改造Java Stream实现的滑动窗口为无重叠模式

核心改造逻辑 #

原来的重叠滑动窗口实现采用步长为1生成索引序列，相邻窗口自然会产生元素重叠。要实现无重叠效果，只需要做两处关键调整：

• 将索引生成的步长调整为与窗口大小相等，让相邻窗口的起点刚好接上上一个窗口的终点，从规则上杜绝元素重叠
• 调整窗口截取的边界逻辑，自动兼容最后一段长度不足窗口大小的尾部内容，不需要单独编写分支判断

具体代码实现 #

替换原方法的内部逻辑即可，完整实现如下：

import java.util.Arrays;
import java.util.List;
import java.util.stream.IntStream;
import java.util.stream.Stream;

public class PreProcess {

    public static <T> Stream<List<T>> createNonOverlappingWindow(List<T> list, int size) {
        // 非法入参直接返回空流
        if (size <= 0 || list.isEmpty()) {
            return Stream.empty();
        }
        // 从索引0开始，每次步进size生成窗口起点，直到索引超出列表范围
        return IntStream.iterate(0, start -> start < list.size(), start -> start + size)
                .mapToObj(start -> {
                    // 结束索引取 起点+size 和 列表长度的较小值，适配尾部长度不足的场景
                    int end = Math.min(start + size, list.size());
                    return list.subList(start, end);
                });
    }

    public static void main(String args[]) {
        Stream<List<Integer>> sw = createNonOverlappingWindow(Arrays.asList(1, 2, 3, 4, 5), 3);
        sw.forEach(System.out::println);
    }
}

运行效果 #

针对输入列表[1,2,3,4,5]、窗口大小3的测试用例，程序输出完全匹配需求：

[1, 2, 3]
[4, 5]

补充说明 #

• 该实现基于list.subList生成原列表的视图，不会额外复制元素，性能与原滑动窗口实现无差异
• 若需要直接舍弃最后一段不足窗口大小的内容，只需要把IntStream.iterate的终止条件从start -> start < list.size()调整为start -> start + size <= list.size()即可
• 原实现中size > list.size()直接返回空流的逻辑已经被移除：当列表长度小于窗口大小时，方法会直接返回包含整个列表的单元素流，符合通用分块逻辑

内容的提问来源于stack exchange，提问作者blue-sky