【剑指Offer】JZ74 和为S的连续正数序列

JZ74 和为S的连续正数序列

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

返回值描述：
输出所有和为S的连续正数序列。序列内按照从小至大的顺序，序列间按照开始数字从小到大的顺序

输入：
9
返回值：
[[2,3,4],[4,5]]

# @param sum int整型 
# @return int整型二维数组
from typing import List
class Solution:
    def FindContinuousSequence(self , sum: int) -> List[List[int]]:
        # write code here
        result = []

        i = 1
        j = 2
        while i < j:
            a = (i+j)*(j-i+1)/2
            if a == sum:
                tmp = []
                for c in range(i,j+1):
                    tmp.append(c)
                result.append(tmp)
                i = i + 1
            elif a < sum:
                j = j+1
            else:
                i = i+ 1
        return result

a = Solution()
print(a.FindContinuousSequence(100))