题目描述
题目:小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
思路:设定两个指针,先分别指向数字1和数字2,并设这两个指针为small和big,对small和big求和,如果和大于目标值,则从当前和中删除small值,并把small值加一,如果和小于目标值,则把big值加一,再把新的big值加入和中。如果和等于目标值,就输出small到big的序列,同时把big加一并加入和中,继续之前的操作。
解题思路
# -*- coding:utf-8 -*-
class Solution:
def FindContinuousSequence(self, tsum):
# write code here
if tsum < 3:
return []
small = 1
big = 2
middle = (tsum + 1) // 2
# 终止条件,最少两个数的和为tsum
cursum = big + small
output = []
# cursum 计算的是[small,big]之间的累加和
while small < middle:
if cursum == tsum:
output.append(list(range(small, big + 1)))
while cursum > tsum and small < middle:
cursum -= small
small += 1
if cursum == tsum:
output.append(list(range(small, big + 1)))
big += 1
cursum += big
return output
class Solution:
def FindContinuousSequence(self, tsum):
# write code here
res = []
# 求[i,...,j]的和
for i in range(1, tsum // 2 + 1):
for j in range(i+1, tsum // 2 + 2):
# 等差序列求和的公式
if (j - i + 1) * (i + j) // 2 == tsum:
res.append(list(range(i, j + 1)))
break
elif (j - i + 1) * (i + j) // 2 > tsum:
break
return res