学习对撞指针的应用
初始时两个指针分别指向第一个元素位置和最后一个元素的位置。每次计算两个指针指向的两个元素之和,并和目标值比较。如果两个元素之和等于目标值,则发现了唯一解。如果两个元素之和小于目标值,则将左侧指针右移一位。如果两个元素之和大于目标值,则将右侧指针左移一位。移动指针之后,重复上述操作,直到找到答案。
使用双指针的实质是缩小查找范围。那么会不会把可能的解过滤掉?答案是不会。假设 numbers [ i ] + numbers [ j ] = target \text{numbers}[i]+\text{numbers}[j]=\text{target} numbers[i]+numbers[j]=target 是唯一解,其中 0 ≤ i < j ≤ numbers.length − 1 0 \leq i<j \leq \text{numbers.length}-1 0≤i<j≤numbers.length−1。初始时两个指针分别指向下标 0 0 0 和下标 numbers.length − 1 \text{numbers.length}-1 numbers.length−1,左指针指向的下标小于或等于 i i i,右指针指向的下标大于或等于 j j j。除非初始时左指针和右指针已经位于下标 i i i 和 j j j,否则一定是左指针先到达下标 i i i 的位置或者右指针先到达下标 j j j 的位置。
如果左指针先到达下标 i i i 的位置,此时右指针还在下标 j j j 的右侧, sum > target \text{sum}>\text{target} sum>target,因此一定是右指针左移,左指针不可能移到 i i i 的右侧。
如果右指针先到达下标 j j j 的位置,此时左指针还在下标 i i i 的左侧, sum < target \text{sum}<\text{target} sum<target,因此一定是左指针右移,右指针不可能移到 j j j 的左侧。
由此可见,在整个移动过程中,左指针不可能移到 i i i 的右侧,右指针不可能移到 j j j 的左侧,因此不会把可能的解过滤掉。由于题目确保有唯一的答案,因此使用双指针一定可以找到答案。
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
left = 0 # 左指针
right = len(numbers) - 1 # 右指针
while left < right:
s = numbers[left] + numbers[right]
if s == target:
return [left+1, right+1]
elif s > target:
right -= 1
else:
left += 1
复杂度分析
- 时间复杂度: O ( n ) O(n) O(n),其中 n n n 是数组的长度。两个指针移动的总次数最多为 n n n 次。
- 空间复杂度: O ( 1 ) O(1) O(1)。