0
点赞
收藏
分享

微信扫一扫

剑指offer-python:58.和为S的连续正数序列

兵部尚输 2022-02-16 阅读 91
python

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

思路:快慢指针。限定条件是,这些序列和不会超过s的一半,s/2 + (s/2 +1)必然大于s,故循环结束条件是:小于s/2。

1)当此时序列的和<s时,快指针+1;

2)当此时序列和<=s时(=s时,先记录序列和),让当前的和减去最小的值来慢慢逼近s。

代码:

class Solution:
    def func(self , s):
        end = s // 2
        lo , hi , cur_sum = 1 ,2 ,3
        res = []
        while lo < end:
            if cur_sum < s:
                hi += 1
                cur_sum += hi
            else:
                if cur_sum == s:
                    res.append(list(range(lo , hi+1)))
                cur_sum -= lo
                lo += 1
举报

相关推荐

0 条评论