剑指 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]]
解题思路:
代码实现:
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