剑指 Offer 57 - II. 和为s的连续正数序列——滑动窗口法

剑指 Offer 57 - II. 和为s的连续正数序列

题目描述

输入一个正整数 target ，输出所有和为 target 的连续正整数序列（至少含有两个数）。

序列内的数字由小到大排列，不同序列按照首个数字从小到大排列。

示例 1：

输入：target = 9
输出：[[2,3,4],[4,5]]
示例 2：

输入：target = 15
输出：[[1,2,3,4,5],[4,5,6],[7,8]]

解题思路

题目意思不难理解，但是写出来我觉得还是有点难度的。。

这里采用的是滑动窗口法

• 窗口怎么扩大或者缩小

  当窗口的和小于 target 的时候，窗口的和需要增加，所以要扩大窗口，窗口的右边界向右移动

  当窗口的和大于 target 的时候，窗口的和需要减少，所以要缩小窗口，窗口的左边界向右移动

  当窗口的和恰好等于 target 的时候，我们需要记录此时的结果。设此时的窗口为 [i, j)，那么我们已经找到了一个 i 开头的序列，也是唯一一个 i 开头的序列，接下来需要找 i+1开头的序列，所以窗口的左边界要向右移动

class Solution_57_11 {
    /**
     * 滑动窗口法
     * @param target
     * @return
     */
    public int[][] findContinuousSequence(int target) {
        int i = 1; //滑动窗口的左边界
        int j = 1; //滑动窗口的右边界
        int sum = 0; //滑动窗口中数字的和
        List<int[]> res = new ArrayList<>();

        while (i <= target/2){
            if (sum < target){
                //右边界向右移动
                sum += j;
                j++;
            }else if (sum > target){
                //左边界向右移动
                sum -= i;
                i++;
            }else {
                //记录结果
                // 若 相等，则：
                //  1、将当前区间数，当做一个结果，存储为数组
                //  2、将 目标数组，存入结果数组
                //  3、左边界 右移，查找 之后的结果
                int[] arr = new int[j - i];
                for (int k = i; k < j; k++) {
                    arr[k - i] = k;
                }
                res.add(arr);
                //左边界向右移动
                sum -= i;
                i++;
            }
        }
        return res.toArray(new int[res.size()][]);
    }
}