面试题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]]
限制:
1 <= target <= 10^5
i 指针表示窗口左边界,j 指针表示窗口右边界。
当 sum < target 时,j 指针右移。
当 sum > target 时,i 指针右移。
当 sum = target 时,窗口内的数组加入 res,i 指针右移。
注意窗口的 range 是左闭右开 [ )。
时间复杂度: O(n)
i每次右移,sum要减去对应之前的i值,这样一旦sum>target 或者等于,就可以通过调整i的位置,不断寻找合适的列表。
class Solution(object):
def findContinuousSequence(self, target):
i, j = 1, 1
sum = 0
res = []
while i <= target // 2:
if sum < target:
# j + 1
sum += j
j += 1
elif sum > target:
# i + 1
sum -= i
i += 1
else:
arr = list(range(i, j))
res.append(arr)
# i + 1
sum -= i
i += 1
return res