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

题目描述

输入一个正整数 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)

猜你喜欢

转载自www.cnblogs.com/flix/p/13382852.html