剑指offer：面试题41——和为s的连续正数序列

最新推荐文章于 2024-07-21 08:48:50 发布

原创最新推荐文章于 2024-07-21 08:48:50 发布 · 406 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

剑指offer

博客围绕剑指offer面试题，探讨和为S的连续正数序列问题。以和为100为例，给出两组连续正数序列。解题思路类似滑动窗口，用两个指针small和big，根据当前和与sum大小关系移动指针，终止条件是small小于和的一半。

剑指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;
    }
};