剑指offer:面试题41——和为s的连续正数序列
题目:
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
思路:类似于滑动窗口,初始定义两个指针指向small和big,如果当前和小于sum则big++扩张(左扩张),否则如果当前和大于sum,则要减小当前集合的总和,small++(右扩张)。且要判断small是否小于和的一半,此为终止条件
class Solution {
public:
void AddVec(int small,int big,vector<vector<int>>&retval)
{
vector<int>seq;
for(int i=small;i<=big;i++)
seq.push_back(i);
retval.push_back(seq);
}
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int>> retval;
if(sum<3) return retval; //只有至少包含1、2的正数序列
int small=1;
int big=2;
int middle=(1+sum)/2; //向上取证难,使得small只能接近,不能取到middle
int curSum=small+big;
while(small<middle) //因为big大于small,所以如果small>middle,两者的和一定大于sum,无意义
{
if(curSum==sum)
AddVec(small,big,retval);
while(curSum>sum&&small<middle)
{
curSum-=small;
small++;
if(curSum==sum)
AddVec(small,big,retval);
}
big++;
curSum+=big;
}
return retval;
}
};