《劍指offer》——和爲s的連續整數序列

題目:輸入一個正整數s,打印出所有和爲s的連續正數序列(至少含有兩個數)。

可以考慮用兩個數small和big分別表示序列的最小值和最大值。首先把small初始化爲1,big初始化爲2。如果從small到big的序列和大於s,可以從序列中去掉最小值,將small的值增大;如果從small到big的序列和小於s,可以將big值增大,讓序列包含新的值。因爲序列至少有兩個數字,所以small的值增加到(s+1)/2爲止。

以s等於9爲例,
這裏寫圖片描述

代碼如下,

/*更新seq*/
void pushSequence(vector<vector<int>> &seq, int small, int big)
{
    vector<int> temp;//定義輔助一維vector
    for(int k = small; k <= big; k++)//將當前序列輸入temp中
        temp.push_back(k);
    seq.push_back(temp);//將temp輸入seq中
}

/*返回值爲多個和爲s的連續整數序列*/
vector<vector<int>> FindContinuousSequence(int sum) 
{
    vector<vector<int>> seq;//定義一個二維vector
    if(sum < 3)
        return seq;//若sum小於3,則返回空的vector
    int small = 1, big = 2;//設置兩個標誌位,分別指向序列的最小值和最大值
    int mid = (sum + 1) / 2;//small的邊界值
    int curSum = small + big;//當前的序列和
    while(small < mid && big < sum)
    {
        while(sum < curSum)//如果當前序列和大於sum,則循環更新當前序列和,並將small指向下一個值
        {
            curSum -= small;
            small++;
        }
        if(curSum == sum)//如果當前序列和等於sum,則將當前的序列輸入seq中
            pushSequence(seq, small, big);
        big++;//如果當前序列和小於sum,則循環將big指向下一個值,並更新當前序列和
        curSum += big;
    }
    return seq;//返回seq
}
發佈了82 篇原創文章 · 獲贊 6 · 訪問量 4萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章