剑指 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]]

我们可以考虑采用滑动窗口的思想：

滑动窗口的重要性质是：窗口的左边界和右边界永远只能向右移动，而不能向左移动。

我们该如何控制窗口的大小：

1. 当 sum > target, 说明窗口的的值多了，那么窗口的左边界向右移动（减去左边的值）；

2. 当 sum < target, 说明窗口的的值少了，那么窗口的右边界向右移动（加上右边的值）；

3. 当 sum = target, 此时记录结果，窗口大小为  [left，right）,为了序列不重复，我们已经找到了以此时left为左边界的序列，下一步就该找left+1的序列, 此时left还是向右移动。

代码贴在下面了

class Solution {
    public int[][] findContinuousSequence(int target) {
        
        // /创建结果数组
        List<int[]> res = new ArrayList<>();

        int left = 1;
        int right = 1;
        int sum = 0;
        while(right < target){
            if(sum < target){
                sum += right;
                right++;
            }
            else if(sum > target){
                sum -= left;
                left++;
            }
            else if(sum == target){
                int[] arr = new int[right-left];
                for(int i = left; i<right; i++){
                    arr[i-left] = i;
                }
                res.add(arr);
                sum -= left;
                left++;
            }
        }
        return res.toArray(new int[0][]);
    }
}