剑指 Offer 57. 和为s的两个数字
输入一个递增排序的数组和一个数字s,在数组中查找两个数,使得它们的和正好是s。如果有多对数字的和等于s,则输出任意一对即可。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[2,7] 或者 [7,2]
示例 2:
输入:nums = [10,26,30,31,47,60], target = 40
输出:[10,30] 或者 [30,10]
限制:
-
1 <= nums.length <= 10^5
-
1 <= nums[i] <= 10^6
来源:力扣(LeetCode)
链接:力扣
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一解
使用双指针 l, r 指向数组的左和右:
-
计算和
s = nums[i] + nums[j]
-
若
s > targets
,则指针 j 向左移动,即执行j = j - 1
-
若
s < targets
,则指针 i 向右移动,即执行i = i + 1
-
若
s = targets
,立即返回数组[nums[i], nums[j]]
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
l, r = 0, len(nums) - 1
while l < r:
s = nums[l] + nums[r]
if s == target:
return nums[l], nums[r]
elif s < target:
l += 1
else: r -= 1
-
时间复杂度:O(N)
-
空间复杂度:O(1)