给你一个**下标从 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] 。
提示:
2 <= numbers.length <= 3 * 104
-1000 <= numbers[i] <= 1000
numbers 按 非递减顺序 排列
-1000 <= target <= 1000
仅存在一个有效答案
双指针
解题思路:
非递减顺序排列,几乎是递增数列,索引从1开始
双指针:
左指针 left = 0
右指针 right = length - 1
判断左右之和:
如果left+right > target,说明搜索范围在右指针左侧,right–
如果left+right < target,说明搜索范围在左指针右侧,left++
如果left+right = target,刚好匹配上,返回[left+1, right+1]
空间复杂度O(1),时间复杂度O(n)
public static int[] twoSum(int[] numbers, int target) {
int n = numbers.length;
int left = 0;
int right = n - 1;
while (left <= right) {
if (numbers[left] + numbers[right] == target) {
return new int[]{left + 1, right + 1};
}
if (numbers[left] + numbers[right] > target) {
right--;
} else {
left++;
}
}
return new int[]{-1, -1};
}
二分查找
解题思路:
非递减顺序排列,几乎是递增数列,索引从1开始
双指针:
左指针 left = 0
右指针 right = length - 1
中间指针 middle = left + (right-left)/2
判断左中之和:
如果left+ middle> target,说明搜索范围在中间指针左侧,right=middle
如果left+ middle < target,说明搜索范围在中间指针右侧,left=middle
如果left+right = target,刚好匹配上,返回[left+1, right+1]
空间复杂度O(1),时间复杂度O(n)
public int[] twoSum3(int[] numbers, int target) {
int n = numbers.length;
int left = 0;
int right = n - 1;
while (left <= right) {
int middle = left + (right - left) / 2;
if (numbers[left] + numbers[right] == target) {
return new int[]{left + 1, right + 1};
}
if (numbers[left] + numbers[middle] > target) {
right = middle;
} else {
left = middle;
}
}
return new int[]{-1, -1};
}
但是这个方法没有经过用例,报错超过限时时长。
优化
在数组中找到两个数,使得它们的和等于目标值,可以首先固定第一个数,然后寻找第二个数,第二个数等于目标值减去第一个数的差。利用数组的有序性质,可以通过二分查找的方法寻找第二个数。为了避免重复寻找,在寻找第二个数时,只在第一个数的右侧寻找。
public static int[] twoSum2(int[] numbers, int target) {
int n = numbers.length;
for (int i = 0; i < n; i++) {
// 默认从第二个数开始查找
int left = i + 1;
int right = n - 1;
while (left <= right) {
int middle = left + (right - left) / 2;
if (numbers[middle] > target - numbers[i]) {
right = middle - 1;
} else if (numbers[middle] < target - numbers[i]) {
left = middle + 1;
} else {
return new int[]{i + 1, middle + 1};
}
}
}
return new int[]{-1, -1};
}