题目
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
思路1
从头遍历列表,用一个字典保存每个数的需要的 另一半: 该数的index, 循环中做判断, 如果数字出现在字典中, 返回结果
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
if not numbers:
return 0, 0
lower_dict = dict()
for index in range(len(numbers)):
if numbers[index] in lower_dict:
return lower_dict[numbers[index]] + 1, index + 1
wanted = target - numbers[index]
lower_dict[wanted] = index
思路2
因为是有序列表,可以使用双指针,头一个尾一个向中间靠拢; 过程中相加和比目标数字大, 尾指针向左挪一位, 比目标数字小, 头指针向右边移动一位; 直至求得目标数字,返回index
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
if not numbers:
return 0, 0
lptr, rptr = 0, len(numbers) - 1
while lptr < rptr:
if numbers[lptr] + numbers[rptr] < target:
lptr += 1
elif numbers[lptr] + numbers[rptr] > target:
rptr -= 1
else:
return lptr + 1, rptr + 1