剑指 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。所以向右边包含一个数。
curSum 还是小于 target ,继续向右扩张。
此时curSum大于target。让左边缩小。
找到了第一个连续的子序列。
让curSum继续向右扩张,curSum小于target 左边缩小
》》》
找到了第二个序列和。 此时子序列中最左边元素4再向右移动就不小于 (target +1) / 2 了。已经没有连续子序列和了,查找完毕。
代码示例:
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;
}
};