题目:小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列?
思路:利用滑动窗口技术,计算窗口[lo,hi]内的和,若与目标值相等,则lo++;若比目标值小,则扩大窗口,即hi++;若比目标值大,则缩小窗口,即lo++;因为初始窗口大小为2,并且当相等时,只有lo++,加后窗口的值肯定比目标值小,所以两个边界都是往右移动;当比目标值大的时候,是lo++不是hi–是因为:本来hi就是从上一个数移动过来才导致的大于目标值,所以选择lo++;当lo追上hi 时,退出循环,后面的序列不必要看,肯定大于目标值,追上它说明区间已经最小了,还不满足,肯定后面的都比目标值大,例如,sum=25,当在[12,13],区间时,是最后一个序列,再往后就追上了,13后的两数和都大于25,故退出。
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
int lo=1;
int hi=2;
vector<vector<int>> res;
while(lo<hi){
vector<int> temp;
int cur=(lo+hi)*(hi-lo+1)/2;//等差数列求和公式
if(cur==sum){//满足条件的序列
for(int i=lo;i<=hi;i++)//存下来
temp.push_back(i);
lo++;
res.push_back(temp);
}
else if(cur<sum){//小于sum
hi++;
}
else{ //大于sum
lo++;
}
}
return res;
}
};