其他算法-041-和为S的连续正数序列(滑动窗口思想)

题目描述

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

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

分析

利用双指针构建的滑动窗口思想，初始两指针指向数组头，和小了增大头指针，和大了增大尾指针，直到头指针大于目标和的一半，终止循环（向上取整）。

代码

# -*- coding:utf-8 -*-
class Solution:
    def FindContinuousSequence(self, tsum):
        # write code here
        
        if tsum<=1:
            return []
        res = []
        
        lowP = 1
        highP = 2
        
        while highP<=(tsum+1)//2:
            curSum = sum(range(lowP,highP+1))
            if curSum<tsum:
                highP += 1
            elif curSum>tsum:
                lowP += 1
            else:
                res.append(list(range(lowP,highP+1)))
                highP += 1
                
        return res