给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
- 返回的下标值(index1 和 index2)不是从零开始的。
- 你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted
感觉之前做过。。。还记得使用双指针做,然后华丽丽地又写了O(N^2)的解法(捂脸哭)
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
ans = []
for i in range(len(numbers)):
for j in range(i+1,len(numbers)):
if numbers[i] + numbers[j] == target:
ans = []
ans.append(i + 1)
ans.append(j + 1)
elif numbers[i] + numbers[j] > target:
break
return ans
果然超出时间限制,看了一下测试样例,满屏的0和9,打扰了。又灰溜溜跑去看了双指针的正确打开方式:让左指针处于第一个元素位置,右指针在最后一个元素的位置,由于是有序序列,所以可以“夹逼”出target的值。numbers[i] + numbers[j] < target ,目前的sum偏小,则让i右移一位,若numbers[i] + numbers[j] > target ,目前的sum偏大,则让j左移一位。直到numbers[i] + numbers[j] = target 得到答案,或者i,j重合,返回 [ ]。这样的算法只需要遍历列表一遍,O(N)。
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
ans = []
i, j =0, len(numbers)-1
while i < j:
if numbers[i] + numbers[j] < target:
i += 1
elif numbers[i] + numbers[j] > target:
j -= 1
elif numbers[i] + numbers[j] == target:
ans = [i+1,j+1]
break
return ans
还看到一个大佬ting-ting-28的哈希表解法:利用字典保存所有差值。
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
dct = {}
for inx, i in enumerate(numbers):
if i in dct:
return [dct[i]+1, inx+1]
dct[target-i] = inx