小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序。
思路:连续数列求和通过知道首尾数字就能知道(等差数列求和)。所以需要俩个指针分别指向头尾数字,让头指针遍历所有数字,固定头指针的同时,移动尾指针寻找在这个头指针的情况下的序列。这种算法复杂度是O(n^2)。另一种思路是根据当前连续序列的和与要求的和的大小关系,移动两个指针的位置。两个指针开始分别指向1、2,一路向左移动,组成一个滑动窗口。
窗口里的和大于要求就头指针向左移,小于就尾指针向左移。
# -*- 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 # high += 1 也行
# 移动指针
elif curSum < tsum:
high += 1
else:
low += 1
return result