两数之和 II - 输入有序数组
难度:中等
给你一个下标从 1 开始的整数数组 numbers,该数组已按 非递减顺序排列,请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] 和 numbers[index2] ,则 1<= index1 < index2 <= numbers.length 。
以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 和 index2。
你可以假设每个输入 只对应唯一的答案 ,而且你 不可以 重复使用相同的元素。
你所设计的解决方案必须只使用常量级的额外空间。
示例 1:
输入:numbers = [2,7,11,15], target = 9
输出:[1,2]
解释:2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。
示例 2:
输入:numbers = [2,3,4], target = 6
输出:[1,3]
解释:2 与 4 之和等于目标数 6 。因此 index1 = 1, index2 = 3 。返回 [1, 3] 。
示例 3:
输入:numbers = [-1,0], target = -1
输出:[1,2]
解释:-1 与 0 之和等于目标数 -1 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。
解法一:二分查找
思路:
假设下标为 i 的数字为两个数中的一个,即可知另外一个数值得大小,令开始为 i + 1,结束为numbers.length - 1,使用二分查找寻找。重复上述操作,依次将 i 向后遍历找到可行解返回即可。
时间复杂度:
O
(
n
l
o
g
n
)
O(nlogn)
O(nlogn)
空间复杂度:
O
(
1
)
O(1)
O(1)
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
for i in range(len(numbers)):
left = i + 1
right = len(numbers) - 1
while left <= right:
mid = int((left + right)/2)
if numbers[mid] == target - numbers[i]:
return [i+1, mid+1]
elif numbers[mid] > target - numbers[i]:
right = mid - 1
else:
left = mid + 1
解法二:双指针
思路:
采用双指针来代替下标,令左指针为第一个下标,右指针为最后一个下标,用左右指针位置上的数之和与目标值进行对比,如果等于目标值,则返回左右指针下标。如果大于则将右指针的往前移动一个,因为有序数组越往右则数越大,往左则越小。如果小于则将左指针的往后移动一个,直到满足返回条件为止。
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
1
)
O(1)
O(1)
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
left = 0
right = len(numbers) - 1
while left < right:
if numbers[left] + numbers[right] == target:
return [left+1, right+1]
elif numbers[left] + numbers[right] > target:
right -= 1
else:
left += 1
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/two-sum-ii-input-array-is-sorted/