167. 两数之和 II - 输入有序数组
给定一个已按照 非递减顺序排列 的整数数组 numbers ,请你从数组中找出两个数满足相加之和等于目标数 target 。
函数应该以长度为 2 的整数数组的形式返回这两个数的下标值。numbers 的下标 从 1 开始计数 ,所以答案数组应当满足 1 <= answer[0] < answer[1] <= numbers.length 。
你可以假设每个输入 只对应唯一的答案 ,而且你 不可以 重复使用相同的元素。
示例 1:
输入:numbers = [2,7,11,15], target = 9
输出:[1,2]
解释:2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
示例 2:
输入:numbers = [2,3,4], target = 6
输出:[1,3]
示例 3:
输入:numbers = [-1,0], target = -1
输出:[1,2]
提示:
2 <= numbers.length <= 3 * 104
-1000 <= numbers[i] <= 1000
numbers 按 非递减顺序 排列
-1000 <= target <= 1000
仅存在一个有效答案
代码1:
直接左右双指针遍历整个数组,和大则右指针减一,和小则左指针加一,直到遍历出结果。
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
n = len(numbers)
left = 0
right = n - 1
while left < right:
sum = numbers[left] + numbers[right]
if sum == target:
return [left + 1, right + 1]
break
elif sum > target:
right -= 1
else:
left += 1
代码2:
采用二分查找方法查找数组中是否存在target-numbers[i]的值。
整个算法相比原始的二分查找仅多了一重for循环和一个==判断。
二分查找算法的精髓是用mid+1和mid-1来代替left和right。
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
n = len(numbers)
for i in range(n):
target0 = target - numbers[i]
left = i + 1
right = n - 1
while left <= right:
mid = (left + right) // 2
if numbers[mid] == target0:
return [i + 1, mid + 1]
elif numbers[mid] > target0:
right = mid - 1
else:
left = mid + 1