问题描述:
输入一个正数s,打印出所有和为s的连续正数序列(至少含有两个数)。例如输入15,由于1+2+3+4+5=4+5+6=7+8=15,所以结果打印出3个连续序列1~5、4~6和7~8。
样例:
- 输入:15
- 输出:[[1,2,3,4,5],[4,5,6],[7,8]]
运用双指针,前提是序列是有序的,设置两个指针 i 和 j, i 向前移动,求和过程是在 i 和 j 之间求和,如果不满足给定的目标值,那么 j 也要向前移动,j 移动的过程在求和的过程中,由于可能涉及多个连续序列和为目标值,因此使用 s-=i; 表示 从 i 下一个位置开始记录,举个例子,比如当前的s=1+2+3+4+5,此时 i=1,那么经过s-=i;之后,s=2+3+4+5,这样就相当于改变了求和空间,并且也记录了下一次求和时需要的部分值。如果满足条件,那么就记录 i 到 j 之间的数。
class Solution {
public:
vector<vector<int> > findContinuousSequence(int sum) {
vector<vector<int>> res;
for(int i=1,j=1,s=1;i<=sum;i++)
{
while(s<sum) s+= ++j ;
if(s==sum&&j-i>0)
{
vector<int> temp;
for(int k=i;k<=j;k++) temp.push_back(k);
res.push_back(temp);
}
s-=i;
}
return res;
}
};
class Solution {
public:
vector<vector<int> > findContinuousSequence(int sum) {
vector<vector<int>> res;
for(int i=1,j=2;j<sum;)
{
int s = (i + j) * (j - i + 1) / 2;
if(s>sum) i++;
else if(s==sum&&i<=j)
{
vector<int> path;
for(int k=i;k<=j;k++) path.push_back(k);
res.push_back(path);
i++;//很重要
}
else j++;
}
return res;
}
};
下一版代码改了好久,原因在于求和等于sum的时候,下一步 i 也需要向后移动,否则下次循环进来一直等于sum,一直运行那段代码知道循环结束。这样是错误的,而且不能在等于sum的时候将初始s=1,那样就相当于从一位置开始,中间不连续直接加了 j ,已经不满足题意,而且这样算下来没有连续的子数组和等于sum了。
class Solution {
public:
vector<vector<int> > findContinuousSequence(int sum) {
vector<vector<int>> res;
int s=1;
for(int i=1,j=2;j<=sum;)
{
if(s<sum)
{
s+=j;
j++;
}
else if(s>sum)
{
s-=i;
i++;
}
else if(s==sum&&i<=j)
{
vector<int> path;
for(int k=i;k<j;k++) path.push_back(k);
res.push_back(path);
s-=i;//很重要
i++;
}
}
return res;
}
};