LeetCode-滑动窗口和为s的连续正数序列

Posted 2020-12-18 Flix

题目描述

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

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

示例：

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

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

说明：

• 1 <= target <= 10^5

题目链接： https://leetcode-cn.com/problems/he-wei-sde-lian-xu-zheng-shu-xu-lie-lcof/

思路

使用滑动窗口来做。初始化 left = 1，right = 2；

• 如果 right<target，循环：
  • 计算 sum = left + right；
  • 如果 sum==target，则将 [left, right] 范围内的数字放入答案中，并将 right 右移一位，也就是 right++，然后 sum+=right（不这样会死循环）；
  • 否则，如果 sum>target，则将窗口左边缩小一位，也就是 sum-=left, left++（这两步的顺序不能错）；
  • 否则，如果 sum<target，则将窗口右边扩大一位，也就是 right++, sum+=right（这两步的顺序不能错）；

代码如下：

class Solution {
public:
    vector<vector<int>> findContinuousSequence(int target) {
        vector<vector<int>> ans;
        if(target==1){
            ans.push_back({1});
            return ans;
        }

        int left = 1;
        int right = 2;
        int sum = left + right;
        while(right<target){
            if(sum==target){
                vector<int> temp;
                for(int i=left; i<=right; i++) temp.push_back(i);
                ans.push_back(temp);
                right++; // 注意要将 right 右移一位，并 sum+=right
                sum += right;
            }else if(sum>target){
                sum -= left;
                left++;
            }else if(sum<target){
                right++;
                sum += right;
            }
        }
        return ans;
    }
};

• 时间复杂度：O(n)
• 空间复杂度：O(1)