剑指Offer刷题笔记——和为S的连续正数序列

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

思路:连续数列求和通过知道首尾数字就能知道(等差数列求和)。所以需要俩个指针分别指向头尾数字,让头指针遍历所有数字,固定头指针的同时,移动尾指针寻找在这个头指针的情况下的序列。这种算法复杂度是O(n^2)。另一种思路是根据当前连续序列的和与要求的和的大小关系,移动两个指针的位置。两个指针开始分别指向1、2,一路向左移动,组成一个滑动窗口。

窗口里的和大于要求就头指针向左移,小于就尾指针向左移。

# -*- coding:utf-8 -*-
class Solution:
    def FindContinuousSequence(self, tsum):
        # write code here
        result = []
        low, high = 1, 2
        while low < high:
            curSum = (low + high) * (high - low + 1) / 2
            # 等于要求就保存起来
            if curSum == tsum:
                tmp = []
                for i in range(low, high+1):
                    tmp.append(i)
                result.append(tmp)
                low += 1  # high += 1 也行
            
            # 移动指针
            elif curSum < tsum:
                high += 1
            else:
                low += 1

        return result

 

 

 

你可能感兴趣的:(剑指Offer刷题笔记——和为S的连续正数序列)