剑指 Offer 57 - II. 和为s的连续正数序列 C++

剑指 Offer 57 - II. 和为s的连续正数序列


输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。

序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。

示例 1:

输入:target = 9
输出:[[2,3,4],[4,5]]

示例 2:

输入:target = 15
输出:[[1,2,3,4,5],[4,5,6],[7,8]]


滑动窗口求解:

按题意,target至少为3才有连续和序列,至少要求两个数。我们让序列从1和2开始。

此时连续子序列和为3,小于target。所以向右边包含一个数。
剑指 Offer 57 - II. 和为s的连续正数序列 C++_第1张图片
curSum 还是小于 target ,继续向右扩张。

剑指 Offer 57 - II. 和为s的连续正数序列 C++_第2张图片

此时curSum大于target。让左边缩小。
剑指 Offer 57 - II. 和为s的连续正数序列 C++_第3张图片

找到了第一个连续的子序列。

剑指 Offer 57 - II. 和为s的连续正数序列 C++_第4张图片

让curSum继续向右扩张,curSum小于target 左边缩小
剑指 Offer 57 - II. 和为s的连续正数序列 C++_第5张图片

》》》
剑指 Offer 57 - II. 和为s的连续正数序列 C++_第6张图片
找到了第二个序列和。 此时子序列中最左边元素4再向右移动就不小于 (target +1) / 2 了。已经没有连续子序列和了,查找完毕。

剑指 Offer 57 - II. 和为s的连续正数序列 C++_第7张图片


代码示例:

class Solution {
public:
    vector<vector<int>> findContinuousSequence(int target) {

        vector<vector<int>> ret;
        if(target < 3)
        {	//target小于3,没有连续子序和
            return ret;
        }

        int left = 1;
        int right = 2;
        //left最大不能大于midlle
        int middle = (target + 1) / 2;
        int curSum = left + right;

        while(left < middle)
        {
            if(curSum == target)
            {
                 vector<int> tmp;
                int tmpleft = left;
                for(;tmpleft<=right;tmpleft++)
                {
                    tmp.push_back(tmpleft);
                }
                ret.push_back(tmp);
            }

            while(curSum > target && left < middle)
            {
                curSum -= left;
                left++;
                
                if(curSum == target)
                {
                     vector<int> tmp;
                    int tmpleft = left;
                    for(;tmpleft<=right;tmpleft++)
                    {
                        tmp.push_back(tmpleft);
                    }   
                    ret.push_back(tmp);
                }
            }

            right++;
            curSum += right;
        }

        return ret;
    }
};

你可能感兴趣的:(剑指 Offer 57 - II. 和为s的连续正数序列 C++)