LeetCode167两数之和2输入有序数组

题目描述

给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。

函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2

说明:

  • 返回的下标值(index1 和 index2)不是从零开始的。
  • 你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。

示例:

1
2
3
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。

解题思路

审题可知,有序的数组,而且不存在重复的元素(只对应唯一的答案),适合使用双数组解题。

使用startend两个指针分别从数组的头部和尾部遍历数组,比较 numbers[start] + numbers[end]tartget 的大小,来移动左右指针。

代码实现

GitHub Python代码地址

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
class Solution(object):
def twoSum(self, numbers, target):
"""
:type numbers: List[int]
:type target: int
:rtype: List[int]
"""
start = 0
end = len(numbers) - 1

while start < end:
result = numbers[start] + numbers[end]
if result == target:
return [start+1, end+1]
elif result > target:
end = end - 1
else:
start = start + 1