python【力扣LeetCode算法题库】面试题57 - II-和为s的连续正数序列(滑动窗口)

面试题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

在这里插入图片描述

发布了833 篇原创文章 · 获赞 217 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/weixin_43838785/article/details/104705290