输入一个递增排序的数组和一个数字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]
解答:
定义两个指针i,j,从一头和一尾开始遍历。如果两数之和小于s,则i加1, 如果两数之和大于s,则j减1
Python代码如下:
def twoSum(nums, target):
i, j = 0, len(nums)-1
while i < j:
s = nums[i] + nums[j]
if s < target:
i += 1
elif s > target:
j -= 1
else:
return [nums[i], nums[j]]
return []
nums = [2,7,11,15]
target = 9
res = twoSum(nums, target)
print(res)