输入一个递增排序的数组和一个数字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)
链接:https://leetcode-cn.com/problems/he-wei-sde-liang-ge-shu-zi-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路1:二分
由于数组是有序的,所以可以遍历每个数,然后进行二分查找,时间复杂度O(nlogn)。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
def erfen(l, r, k):
while l <= r:
m = (l + r) >> 1
if nums[m] == k:
return k
elif nums[m] > k:
r = m - 1
else:
l = m + 1
return -1
length = len(nums)
for i in range(length-1):
f = erfen(i+1, length-1, target - nums[i])
if f != -1:
return [nums[i], f]
思路2:双指针
因为有序,所以可以利用双指针进行双向查找。前指针i指向第一个数,后指针j查找target - nums[i]的值,如果该值大于查找在,这j–,往前推进。其正确性很容易证明,因为如果存在多匹对项,它们的分布一定是:
…a1.a2.a3…b3.b2.b1…
因为a1+b1=a2+b2=a3+b3,所以如果a1<a2,那么b1>b2。
时间复杂度:O(n)。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
# 双指针
i = 0
j = len(nums) - 1
while i < j:
if nums[j] == target - nums[i]:
return [nums[i], nums[j]]
elif nums[j] > target - nums[i]:
j -= 1
else:
i += 1