[剑指offer]-和为S的连续正数序列

题目描述

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

输出描述:

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

有了解决前面问题的经验，我们也考虑用两个数small和big分别表示序列的最小值和最大值。首先把small初始化为1，big初始化为2。如果从small到big的序列的和大于s，则可以从序列中去掉较小的值，也就是增大small的值。如果从small到big的序列的和小于s，则可以增大big，让这个序列包含更多的数字。因为这个序列至少要有两个数字，我们一直增加small到（1+s）/2为止。

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int> > result;
        if(sum <3)
            return result;
        int left = 1;
        int right = 2;
        int mid = (1+sum)/2;
        while(left < mid){
            if(sum == calculate(left,right)){
                vector<int> temp;
                for(int i=left;i<=right;i++)
                    temp.push_back(i);
                result.push_back(temp);
                right++;
            }
            else if(sum > calculate(left,right))
                right++;
            else
                left++;
        }
        return result;
    }
    long calculate(int a,int b){
        long sum=0;
        for(int i=a;i<=b;i++)
            sum += i;
        return sum;
    }
};