题目:
输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。
序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。
示例 1: 输入:target = 9 输出:[[2,3,4],[4,5]]
思路:
- 双指针法(滑动窗口)
参考链接- 双指针初始值:left=1,right=2
这里直接用指针的值来代表序列中的数字。 - 双指针变化规则:根据【左右指针之间滑动窗口内的数值和s】与目标值【target】的大小关系来变化
s < target,right指针右移一位,滑动窗口变大,数值和s=s+right
s==target,此时滑动窗口内的正整数序列符合要求,①将其记录到最终结果中去。注意,可能还存在其他的正整数序列符合要求(如target=9时,滑动窗口移动至[2,3,4]时s == target,符合要求,但后面的序列[4,5]也符合要求,即滑动窗口还可以右移),②数值和s=s-left,left指针右移一位
s > target,left指针右移一位,滑动窗口变小,数值和s=s-left。注意,s-=left
操作须在left+=1
前。 - 循环终止条件:
while left<right
题目要求连续正整数序列至少含有两个数,当滑动窗口移动到【只有两个数(right-left=1)且s==target】的情况时,证明此后再无符合要求的序列(越往后数越大),此时进入【s == target】的语句块,left指针右移一位,此时left == right,跳出循环,返回最终的结果。
- 双指针初始值:left=1,right=2
代码:
class Solution:
def findContinuousSequence(self, target: int) -> List[List[int]]:
left,right=1,2
s=3
res=[]
while left<right:
if s<target:
right+=1
s+=right
elif s==target:
res.append(list(range(left,right+1)))
s-=left
left+=1
elif s>target:#可直接写成else
s-=left
left+=1
return res
可以发现s==target情况下除了记录值以外,其余操作和s>target下的操作完全相同。整理后:
class Solution:
def findContinuousSequence(self, target: int) -> List[List[int]]:
left,right=1,2
s=3
res=[]
while left<right:
if s==target:#这里的if语句和下面的if-else语句独立,同时进行这两个判断
res.append(list(range(left,right+1)))
if s>=target:#这里的等号可以去掉,s==target时,左/右指针右移均可
s-=left
left+=1
else:
right+=1
s+=right
return res