题目来自LeetCode,链接:two-sum-ii。具体描述为:给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。说明:
- 返回的下标值(index1 和 index2)不是从零开始的。
- 你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
这跟前面的两数之和问题很相似,不过这里多了一个数组有序的前提。当前可以用前面的方法去做,不过由于数组有序,使得我们可以用更高效的方法。而这其实就是前面三数之和里出现过的子问题,可以直接套用里面左右双指针的方法,也就是初始化左指针为数组开头,右指针为数组末尾,然后根据左右指针对应数组之和与target的关系决定移动哪个指针,相等时直接返回,小于时右移左指针,大于时左移右指针。时间复杂度为 O ( N ) O(N) O(N)。
JAVA版代码如下:
class Solution {
public int[] twoSum(int[] numbers, int target) {
int left = 0, right = numbers.length - 1;
while (left < right) {
int sum2num = numbers[left] + numbers[right];
if (sum2num == target) {
return new int[] {left + 1, right + 1};
}
else if (sum2num < target) {
++left;
}
else {
--right;
}
}
return null;
}
}
提交结果如下:
![](https://i-blog.csdnimg.cn/blog_migrate/63289b6cdb7e71acd06ecc43e243f5ae.png)
Python版代码如下:
class Solution:
def twoSum(self, numbers: List[int], target: int) -> List[int]:
left = 0
right = len(numbers) - 1
while left < right:
sum2num = numbers[left] + numbers[right]
if sum2num == target:
return [left + 1, right + 1]
elif sum2num < target:
left += 1
else:
right -= 1
return []
提交结果如下:
![](https://i-blog.csdnimg.cn/blog_migrate/a3dda4005492fa93c1d9db715c63b390.png)