剑指 Offer 57 - II. 和为s的连续正数序列 | 算法

80 阅读1分钟

剑指 Offer 57 - II. 和为s的连续正数序列

输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。

序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。

示例 1:

输入:target = 9
输出:[[2,3,4],[4,5]]
示例 2:

输入:target = 15
输出:[[1,2,3,4,5],[4,5,6],[7,8]]

解题思路

www.bilibili.com/video/BV1RL…

代码实现

class Solution:
    def findContinuousSequence(self, target: int) -> List[List[int]]:
        res = []
        lo, hi = 1, 2
        curSum = lo + hi
        while hi < target:
            if curSum > target:
                curSum -= lo
                lo += 1
            elif curSum < target:
                hi += 1
                curSum += hi
            else:
                res.append([i for i in range(lo, hi + 1)])
                curSum -= lo
                lo += 1
        return res