剑指offer-41 -- 和为S的连续正数序列 - C++

题目描述

小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!

返回值描述:

输出所有和为S的连续正数序列。序列内按照从小至大的顺序，序列间按照开始数字从小到大的顺序

方法一：暴力方法

从第一个数（也就是1）开始（记为i，第一层循环），每向右移一位（记为j，第二层循环），计算一下中间所有数字的和是否为sum；

如果 和==sum，则符合要求；如果 和>sum，说明已经超过了sum，不需要再考虑j后面的数字，直接break本次j的循环，直接进入下一个i

注意：中间计算所有数字的和时，需要遍历从i到j，所以是又一次循环，所以时间复杂度为N的三次方

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int>> ret;
        for(int i=1; i<=sum/2; i++){
            for(int j=i+1; j<sum; j++){
                int tmp = 0;
                vector<int> ans;
                for(int k=i; k<=j ;k++){
                    tmp += k;
                    ans.push_back(k);
                }
                if(sum ==tmp){
                    ret.push_back(ans);
                }
                else if(tmp > sum){
                    break;
                }else{
                    ans.clear();
                }
            }
        }
        return ret;
    }
};

时间复杂度：O(N^3)
空间复杂度：O(1)

 

方法二：前缀和

对于求一个区间和，一贯的优化技巧是使用前缀和。比如：

index	0	1	2	3	4	5
value	0	1	2	3	4	5
sum	0	1	3	6	10	15

 

sum[i]表示前i个数的和。比如sum[1] = 1,表示前一个数的和为1，sum[2] = 3, 表示前2个数的和为3.现在我们要求区间[2,4]表示求第2,3,4个数的和，就等于sum[4] - sum[1] = 9
代码中我们用一个变量来模拟这个前缀和。

本题的思路：用一个tmp记录从i到j的和，如果刚好和sum相等，则从i到j是一个序列；如果和超过了sum，说明从i开始不可能有符合要求的序列，则直接跳出j的循环，直接i++

同样的道理，在找到满足要求的序列时，需要遍历从i到j来存放结果，但只有满足条件才会遍历

代码如下：

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int>> ret;
        int tmp = 0;
        for(int i=1; i<=sum/2; i++){
            for(int j=i; j<sum; j++){
                tmp += j;
                if(sum == tmp){
                    vector<int> ans;
                    for(int k=i; k<=j; k++)
                        ans.push_back(k);
                    ret.push_back(ans);
                }
                else if(tmp > sum){
                    tmp = 0;
                    break;
                }
            }
        }
        return ret;
    }
};

时间复杂度：O(N^2)
空间复杂度：O(1)

方法三：滑动窗口(参考牛客)

知识补充：

1. 什么是滑动窗口？
   顾名思义，首先是一个窗口，既然是一个窗口，就需要用窗口的左边界i和右边界j来唯一表示一个窗口，其次，滑动代表，窗口始终从左往右移动，这也表明左边界i和右边界j始终会往后移动，而不会往左移动。
   这里我用左闭右开区间来表示一个窗口。比如

1. 滑动窗口的操作

• 扩大窗口，j += 1
• 缩小窗口，i += 1
  算法步骤：

1. 初始化，i=1,j=1, 表示窗口大小为0
2. 如果窗口中值的和小于目标值sum， 表示需要扩大窗口，j += 1
3. 否则，如果狂口值和大于目标值sum，表示需要缩小窗口，i += 1
4. 否则，等于目标值，存结果，缩小窗口，继续进行步骤2,3,4

这里需要注意2个问题：

1. 什么时候窗口终止呢，这里窗口左边界走到sum的一半即可终止，因为题目要求至少包含2个数
2. 什么时候需要扩大窗口和缩小窗口？解释可看上述算法步骤。

代码如下：

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int>> ret;
        int l = 1, r = 1;
        int tmp = 0;
        while(l <= sum/2){
            if(tmp < sum){
                tmp += r;
                r++;
            }
            else if(tmp > sum){
                tmp -= l;
                l++;
            }
            else{
                vector<int> ans;
                for(int k=l; k< r; k++)
                    ans.push_back(k);
                ret.push_back(ans);
                tmp -= l;
                l++;
            }
        }
        return ret;
    }
};

时间复杂度：O(N)
空间复杂度：O(1)