题目 和为S的连续正数序列
描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
返回值描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
示例1
输入:
9
返回值:
[[2,3,4],[4,5]]
解1:双向队列
- 设置一个双向队列q
- 从1-((sum+1)/2)进行遍历加和s:(因为是连续加和,所以最大不会超过sum的一半)
如果和比 s > sum:从小的数字逐渐抛出,即从q的头部抛出,并将和s减去抛出的元素。直到:s<=sum - 抛出后,判断s和sum是否相等,相等则将deque所有元素放入一个vector序列中,并存入结果集
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int>> res;
if(sum == 0 || sum == 1) return res;
deque<int> q;
int s = 0;
int val = 0;
for(int i=1 ; i<=(sum+1)/2 ; i++)
{
q.push_back(i);
s += i;
while(!q.empty() && s > sum)
{
val = q.front();
q.pop_front();
s -= val;
}
if(s == sum)
{
vector<int> vec(vector<int>{q.begin() , q.end()});
res.push_back(vec);
}
}
return res;
}
};
解2:双指针
通过两个指针p1 , p2指向头、尾,求p1 - p2的和s
- s > sum:p1++;
- s < sum:p2++;
- s = sum:p1 - p2 放入vector中,放入结果集。p1++
注意:相等后,一定要p1++或者p2++,否则会进入死循环。
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int>> res;
if(sum == 0 || sum == 1) return res;
int p1 = 1;
int p2 = 2;
while(p1 < p2)
{
int s = (p1+p2)*(p2-p1+1)/2;
if(s < sum)
p2++;
else if(s > sum)
p1++;
else
{
vector<int> vec;
for(int i=p1 ; i<=p2 ; i++)
{
vec.push_back(i);
}
res.push_back(vec);
p1++;
}
}
return res;
}
};