题目描述:
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
思路:
三个变量,cur_sum,l,r初始值都为1
如果cur_sum小于目标值,则r+1,cur_sum+r+1
如果cur_sum大于目标值,则l+1,cur_sum-l
代码:
class Solution:
def FindContinuousSequence(self, tsum):
# write code here
l, r, cur_sum = 1, 1, 1
res = []
while r < tsum:
if cur_sum == tsum:
res.append(list(range(l, r + 1)))
l += 1
cur_sum -= l - 1
if cur_sum < tsum:
r += 1
cur_sum += r
else:
l += 1
cur_sum -= l - 1
return res