题目要求
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
解题分析
有了面试题57的基础【剑指offer】面试题57 和为S的两个数字,
我们很容易想到还是用两个指针去做,但是和57题不一样的是,我们没办法同时从头尾(没有已知数组)开始遍历,所以我们这次同时从头开始(原题说正数序列所以从1开始)设置两个变量small = 1, big = 2. 这时候我们有了一个区间[small, big],我们只要计算这个区间的和然后和sum进行比较即可,只可能有三种情况:
(1)和等于sum, 返回[small, big]之间的序列
(2)大于sum, 那么需要减小一个数,所以把[small, big] 的区间变成 [small+1, big]
(3)小于sum, 那么需要增加一个数,所以把[small, big] 的区间变成 [small, big+1]
注意, 题目要求序列至少包括两个数字,所以small一定要小于(1+sum)/2
还有很关键的一点就是,增加small和big后计算序列和的顺序问题,因为增加small是因为原序列的和大于目标值,所以要先减去small,再对small加一,再去判断和目标值的关系。而增加big是因为原序列的和小于目标值,所以要先对big加一再去对整体加和。
主要代码c++
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int>>SumResult;
if(sum < 3)
return SumResult;
int small = 1;
int big = 2;
int middle = (1+sum)>>1;
int CountSum = small + big;
while(small < middle)
{
if(CountSum == sum)
SumResult = Recordnum(small,big,SumResult);
// 如果现在的和大于所求值,那么就减少数字个数[small+1, big]
while(CountSum>sum && small < middle)
{
CountSum-=small;
small++;
if(CountSum == sum)
SumResult = Recordnum(small,big,SumResult);
}
// 如果现在的和小于所求值,那么增加数字个数[small, big+1]
big++;
CountSum+=big;
}
return SumResult;
}
private:
vector<vector<int>>Recordnum(int begin, int end, vector<vector<int>>&SumResult)
{
vector<int>Eachsumm; // 记录每一组的序列
for(int i = begin; i <= end; ++i)
{
Eachsumm.push_back(i);
}
SumResult.push_back(Eachsumm);
return SumResult;
}
};
总结
题目稍微有一点点绕,就是在增加和减少数字那一块,不同于第57题我们对单个数字的增加或减小,对于序列中的数字考虑增加和减少总的数字个数。
[small-1, big] 和 [small, big+1] 很关键!