输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。
序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。
输入:target = 9 输出:[[2,3,4],[4,5]]输入:target = 15 输出:[[1,2,3,4,5],[4,5,6],[7,8]]
卧槽?这是简单题。。。我发现我可能没有脑子……
参考了一下别人的思路,整理出来一份自己的思考过程。
思路:
题目要求和为targe的连续正整数,那么我们可以设想一个区间。
如果这个区间内的所有数字之和等于target,那么这个区间内的所有数字就是我们要找的某一组序列。
如果这个区间内的所有数字之和小于target,那么右边索引往后走一个,直到找到等于target的一组序列。
如果这个区间内的所有数字之和大于target,那么左边索引往前走一个,直到找到等于target的一组索引。
循环结束条件:左边索引与右边索引重合。(此时只有一个数字,不符合题目要求)
以最小的两个正整数为起始条件,我进行了如下思考。
当s==target的时候移动左边界还是右边界?我发现,如果移动右边界的话会出现无法结束循环的情况,一直一直往后走,到了i==j也不会结束循环。如下图所示:
如果s==target移动左边界的话,在i==j处结束循环,满足要求。如下图所示:
然后是关于s的计算,一开始我想到的是用range循环从i到j+1来计算出s,然后又觉得这样每次移动了i或者j都需要循环计算,时间、内存消耗都很大。
后来想到:
每次向后移动i,这组数据中就少了原来i所在位置的数字。
每次向后移动j,这组数据中就多了j现在所在位置的数字。
步骤:
1.循环条件:i<j
2.循环:计算s,与target比较
1.s>=target,i+=1,s=s-i
2.s<target,j+=1,s=s+j
s==target,这组数据记录到列表里。
代码实现如下
class Solution:
def findContinuousSequence(self, target: int) -> List[List[int]]:
i,j,s=1,2,3
result=[]
while i<j:
if s==target and i!=j:
result.append(list(range(i,j+1)))
if s>=target:
# 先计算s再移动,这时候的i才是未移动的
s-=i
i += 1
elif s<target:
# 先移动j,这时候的j才是加入序列后的
j+=1
s+=j
return result