
一、前言
本系列文章为《剑指Offer》刷题笔记。
刷题平台:牛客网
书籍下载:共享资源
二、题目
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序。
1、思路
这道题还是蛮简单的。
设定两个指针,一个指向第一个数,一个指向最后一个数,在此之前需要设定第一个数和最后一个数的值,由于是正数序列,所以可以把第一个数设为1,最后一个数为2(因为是要求是连续正数序列,最后不可能和第一个数重合)。下一步就是不断改变第一个数和最后一个数的值,如果从第一个数到最后一个数的和刚好是要求的和,那么把所有的数都添加到一个序列中;如果大于要求的和,则说明从第一个数到最后一个数之间的范围太大,因此减小范围,需要把第一个数的值加1,同时把当前和减去原来的第一个数的值;如果小于要求的和,说明范围太小,因此把最后一个数加1,同时把当前的和加上改变之后的最后一个数的值。这样,不断修改第一个数和最后一个数的值,就能确定所有连续正数序列的和等于S的序列了。
注意:初中的求和公式应该记得吧,首项加尾项的和乘以个数除以2,即sum = (a + b) * n / 2。
2、代码
C++:
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > result;
// 高位指针和低位指针
int phigh = 2, plow = 1;
// 终止条件是phigh等于sum
while(phigh > plow){
// 当前和,使用求和公式s = (a+b) * n / 2
int curSum = (plow + phigh) * (phigh - plow + 1) >> 1;
if(curSum < sum){
phigh++;
}
if(curSum == sum){
vector<int> temp;
for(int i = plow; i <= phigh; i++){
temp.push_back(i);
}
result.push_back(temp);
plow++;
}
if(curSum > sum){
plow++;
}
}
return result;
}
};
Python:
# -*- coding:utf-8 -*-
class Solution:
def FindContinuousSequence(self, tsum):
# write code here
result = []
low, high = 1, 2
while low < high:
curSum = (low + high) * (high - low + 1) / 2
if curSum == tsum:
tmp = []
for i in range(low, high+1):
tmp.append(i)
result.append(tmp)
low += 1
elif curSum < tsum:
high += 1
else:
low += 1
return result
来源:
https://cuijiahua.com/blog/2018/01/basis_41.html
