题目描述:小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列?
参考书解,使用双指针创造滑动窗口,计算窗口的和与sum判断。一个小技巧:等差数列公式:Sn = (a1 + an) * n / 2; 由于是连续的正整数序列,因此 n = (an - a1 + 1)。
AC代码:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > result;
int plow = 1, phigh = 2;
while(plow < phigh)
{
int sub_sum = (plow + phigh) *(phigh - plow+1) / 2;
if(sub_sum == sum)
{
vector<int> vec;
for(int i = plow;i<=phigh;i++)
{
vec.push_back(i);
}
result.push_back(vec);
plow++;
}
else if(sub_sum < sum)
phigh++;
else
{
plow++;
}
}
return result;
}