和为s的连续整数序列【python】

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


使用双指针的方法:small 和 big,如果序列和小于sum,那么big+1,如果大于sum,那么small + 1,直到small等于s / 2
和为s的连续整数序列【python】
为了防止重复计算和,可以在每次遍历的时候,sum加上或者减去相应的数字

class Solution:
    def FindContinuousSequence(self, tsum):
        # write code here
        if tsum < 3:
            return []
        small, big = 1, 2
        results = []
        sum = small + big
        while small < tsum / 2:
            if sum < tsum:
                big += 1
                sum += big
            elif sum > tsum:
                sum -= small
                small += 1
            else:
                results.append(list(range(small, big + 1)))
                sum -= small
                small += 1
        return results
print(Solution().FindContinuousSequence(100))

输出:[[9, 10, 11, 12, 13, 14, 15, 16], [18, 19, 20, 21, 22]]