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

输入一个正整数 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

class Solution:
    def findContinuousSequence(self, target: int) -> List[List[int]]:
        # 两个指针i,j,如果R(i,j)=Ri+....+(Rj-1)+Rj > target,则i=i+1
        # 如果R(i,j)=Ri+....+(Rj-1)+Rj > target,则j=j+1
        # 因为结果列表至少有两个数,则j-i>=1
        res_lists = []

        i = 1
        j = 2
        while i < j:
            res = 0

            for k in range(i,j):
                res += k

            # 如果i+(i+1)+(i+2)....+(j - 1)+j 小于target,则j+=1
            if res < target:
                j += 1
            # 如果i+(i+1)+(i+2)....+(j - 1)+j 大于target,则i+=1
            
            elif res > target:
                i += 1
            else:
                # 至少含有两个数
                if j - i > 1:
                    res_lists.append(list(range(i,j)))
                    # print(res_lists)
                j += 1
        return res_lists
            

你可能感兴趣的:(算法,python,算法)