剑指offer 和为S的连续正数序列 Python

题目描述

小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!

输出描述:

输出所有和为S的连续正数序列。序列内按照从小至大的顺序，序列间按照开始数字从小到大的顺序
题目链接

解题思路

定义一个asum，当asum<tsum时，则序列长度加1，否则减少序列长度。当起始数字大于tsum一半时，则跳出循环

# -*- coding:utf-8 -*-
class Solution:
    def FindContinuousSequence(self, tsum):
        # write code here
        if tsum <=2:return [] 
        asum,start,end = 0,1,1
        array = []
        while start <= tsum :
            if asum==tsum and end > start+1 : array.append([_ for _ in range(start,end)])
            if asum < tsum:
                asum += end
                end += 1
            else :
                asum -= start
                start += 1
        return array