题目描述:
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
解题思路:
1、定义两个指针i([1, tsum//2])和j([1, tsum])
2、通过两个while循环找到i到j的值累加和等于tsum的i和j,根据下面的公式计算即可:
temp = int(j* (j + 1) /2 - i* (i - 1) /2 )
3、不断更新i和j,直到不符合条件为止。
完整代码:
# -*- coding:utf-8 -*-
class Solution:
def FindContinuousSequence(self, tsum):
# write code here
'''
1、25 ms 5763K
2、运行时间:30ms
占用内存:5724k
'''
if tsum <= 2:
return []
result = []
i = 1
max_ = tsum //2
while i <= max_:
j = i
entry = True
while j < tsum and entry:
temp = int(j* (j + 1) /2 - i* (i - 1) /2 )
if temp == tsum:
result.append(list(range(i , j + 1)))
elif temp > tsum:
entry = False
j += 1
i += 1
if len(result) == 0:
return []
return result