给你一个下标从 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] 。
假设数组 numbers = [1,3,4,5,7,8], target = 10
public int[] twoSum(int[] numbers, int target) {
for (int i = 0; i < numbers.length; ++i) {
// 计算差值
int diff = target - numbers[i];
for (int j = i + 1; j < numbers.length; j++) {
if (numbers[j] == diff) {
// 题目要求下标从1开始,故需+1
return new int[]{i + 1, j + 1};
}
}
}
return new int[]{-1, -1};
}
而在一个有序数组中找目标值,可以利用二分查找:
int target = ;
int low = , high = ;
while (low <= high) {
// 找到中间位置
int mid = (high - low) / 2 + low;
if (numbers[mid] == target) {
// 找到目标值
} else if (numbers[mid] > diff) {
// 中间位置的值大于目标值,则从中间位置往小的方向继续查找
high = mid - 1;
} else {
// 中间位置的值大于目标值,则从中间位置往大的方向继续查找
low = mid + 1;
}
}
完整算法:
public int[] twoSum(int[] numbers, int target) {
// 因为数组是非降序,那么从 index = 0 开始遍历
for (int i = 0; i < numbers.length; ++i) {
// 差值
int diff = target - numbers[i];
// 二分查找目标值
int low = i + 1, high = numbers.length - 1;
while (low <= high) {
int mid = (high - low) / 2 + low;
if (numbers[mid] == diff) {
return new int[]{i + 1, mid + 1};
} else if (numbers[mid] > diff) {
high = mid - 1;
} else {
low = mid + 1;
}
}
}
return new int[]{-1, -1};
}