leetcode167. 两数之和 II - 输入有序数组
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
- 返回的下标值(index1 和 index2)不是从零开始的。
- 你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
方法一:二分查找
思路:
最容易想到的是暴力法,双层循环两两对比即可,但是这种方法的时间复杂度为O(n^2),我们对这个进行优化,遍历数组,作为第一个数,然后,再在数组后边部分进行二分查找,查找target-nums[i],二分查找找不到的话,继续遍历下一个数,再二分......这样可以将时间复杂度降为O(nlogn)。
代码:
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
n = len(numbers)
def binarysearch(left,right,target): #二分查找的代码,查找到的话,返回下标
while left < right:
mid = left + (right-left)//2
if numbers[mid] < target:
left = mid + 1
else:
right = mid
if numbers[left] == target:
return left
else:
return -1
temp = 0
for i in range(n-1): #遍历数组,作为第一个数,不需要遍历到最后一个数,因为后面没有数
temp = binarysearch(i+1,n-1,target-numbers[i])
if temp != -1:
return [i+1,temp+1] #查找到,返回的是第几个,所以下标要+1
结果:
方法二:双指针
思路:
上面的方法本质上与暴力法没差,只是进行了搜索的优化,下面使用双指针的方法来将时间复杂度降到O(n)。
左右指针首先置于头和尾,计算这两个数之和,如果大于target,则右指针左移,如果小于target,则左指针右移,不断更新和,直至找到等于target情况。
代码:
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
n = len(numbers)
left = 0 #初始化左右指针
right = n-1
summ = 0
while left < right:
summ = numbers[left] + numbers[right] #summ维护两个指针指向的数的和
if summ == target:
return [left+1,right+1]
break
elif summ < target:
left += 1
else:
right -= 1