题目描述:
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
解题思路:
双指针法,p1和p2指向某一段区间,如果和为sum则表示找到一组连续正数序列, 如果小于sum,则移动p2,加入更大的数,如果大于sum则移动p1。
Demo:
class Solution {
public:
vector<vector<int>> vec_out;
vector<vector<int> > FindContinuousSequence(int sum) {
if (sum >= 3){
int p1 = 1, p2 = 2, mid = (sum + 1) / 2;
while (p1 < mid){
int cur_sum = (p1 + p2) * (p2 - p1 + 1) / 2;
if (cur_sum == sum){
vector<int> vec_tmp;
for (int i = p1; i <= p2; ++i){
vec_tmp.push_back(i);
}
vec_out.push_back(vec_tmp);
++p2;
}
else if (cur_sum < sum){
++p2;
}
else{
cur_sum -= p2;
++p1;
}
}
}
return vec_out;
}
};