和为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 <= target <= 10^5

思路

利用题目中给定的序列连续的特点,我们想到等差数列求和公式.

我们使用数学公式sum=(r-l+1)(l+r)/2这个公式来计算,如果最终的sum<target,那么说明r应该向后移动,如果sum>target,l++,如果相等,还是应该使得l++,为什么呢?因为只有保证l++之后,这个滑动窗口才能向右滑行,寻找下一次符合条件的机会.

代码

#include<vector>
using namespace std;

class Solution {
public:
    vector<vector<int>> findContinuousSequence(int target) {
        vector<vector<int>> res;
        vector<int> vec;
        for (int l = 1, r = 2; l < r;) {
            int sum = (r - l + 1) * (l + r) / 2;
            if (sum == target) {
                vec.clear();
                for (int i = l; i <= r; ++i) {
                    vec.push_back(i);
                }
                res.push_back(vec);
                l++;
            }
            else if (sum > target) l++;
            else r++;
        }
    }
};