【劍指offer】和爲S的連續正數序列

題目描述

小明很喜歡數學,有一天他在做數學作業時,要求計算出9~16的和,他馬上就寫出了正確答案是100。但是他並不滿足於此,他在想究竟有多少種連續的正數序列的和爲100(至少包括兩個數)。沒多久,他就得到另一組連續正數和爲100的序列:18,19,20,21,22。現在把問題交給你,你能不能也很快的找出所有和爲S的連續正數序列? Good Luck!

輸出描述

輸出所有和爲S的連續正數序列。序列內按照從小至大的順序,序列間按照開始數字從小到大的順序

思路

這道題的特殊之處在於所有的序列都是連續的,因此就容易想到數列的求和公式,只要知道整個序列中的第一個數和最後一個數就可以知道整個序列的和是否爲sum,

然而問題來了,序列中的第一個數和最後一個數應該怎麼得到呢,

這時我們就應該想到用我們常見的雙指針滑動窗口來解決這種問題,

如果窗口內的和小於sum,我們就向後移動右指針;如果大於sum,就移動窗口的左指針。

下面結合代碼捋一遍

代碼

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int>> res;
        //利用雙指針滑動窗口
        int l = 1;//左指針
        int r = 2;//右指針
        while(l < r){  //注意序列至少包括兩個數
            int ans = (l+r)*(r-l+1)/2;//連續序列利用求和公式
            if(ans == sum){
                vector<int> t;
                for(int i = l; i <= r; i++)
                    t.push_back(i);
                res.push_back(t);
                l++;
                r++;
            }
            else if(ans < sum)
                r++;
            else
                l++;
        }
        return res;
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章