剑指offer : 和为S的来连续正数数列
题目描述:小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
连续的正数序列就是一个等差数列,那么可以利用一个等差数列的求和公式来计算当前数列的和。
等差数列求和:(首项+末项)* 项数 / 2;
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum)
{
vector<vector<int> > result;
int first = 1;
int end = 2;
int cur_sum = 0;
while (first < end)
{
cur_sum = (first + end)*(end - first + 1) / 2;
if (cur_sum < sum)
{
end++;
}
else if (cur_sum == sum) // 相等的情况下 就将当前正数队列的值存入
{
vector<int> cur_result;
for (int i = first; i <= end; i++)
{
cur_result.push_back(i);
}
result.push_back(cur_result);
first++;
}
else
first++;
}
return result;
}
};