题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
方法一:以small、big表示序列的最小值和最大值,根据情况更新small、big、curSum值
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > res;
if(sum < 3)
return res;
int small = 1, big = 2;
int mid = (sum + 1)/2;//序列至少含两个数,求得中间值最大值
int curSum = small + big;//更新序列和,每次只增减一个数字,避免重复计算,提升代码效率。
while(small < mid)
{
if(curSum == sum)
insertRes(small, big, res);
while(curSum > sum && small < mid)
{
curSum -= small;
small ++;
if(curSum == sum)
insertRes(small, big, res);
}
big ++;
curSum += big;
}
return res;
}
void insertRes(int begin, int end, vector<vector<int> > &res)
{
vector<int> tmp;
for(int i = begin; i <= end; ++i)
tmp.push_back(i);
res.push_back(tmp);
}
};
方法二:由于是连续的,差为1的一个序列,那么求和公式是(a0+an)*n/2
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > res;
if(sum < 3)
return res;
int small = 1, big = 2;
int mid = (sum + 1)/2;//序列至少含两个数,求得中间值最大值
//int curSum = small + big;
while(small < mid)
{
intcurSum = (small + big) * (big - small + 1)/2;
if(curSum < sum)
big ++;
if(curSum == sum)
{
insertRes(small, big, res);
small ++;
}
insertRes(small, big, res);
if(curSum > sum)
small ++;
}
return res;
}
void insertRes(int begin, int end, vector<vector<int> > &res)
{
vector<int> tmp;
for(int i = begin; i <= end; ++i)
tmp.push_back(i);
res.push_back(tmp);
}
};