题目链接:167. 两数之和 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] 。
提示:
- 2 <= numbers.length <= 3 * 104
- -1000 <= numbers[i] <= 1000
- numbers 按 非递减顺序 排列
- -1000 <= target <= 1000
- 仅存在一个有效答案
代码(c++):
方案一:二分查找
在数组中找到两个数,使得它们的和等于目标值,可以先固定第一个数,然后再去寻找第二个数,第二个数等于目标值减去第一个数的差。利用数组的非递减顺序排列的性质,可以通过二分查找的方法寻找第二个数。为了避免重复寻找,在寻找第二个数的时候,只在第一个数的右侧寻找。
//二分查找
//先固定一个元素nums[i],然后二分查找target - nums[i]
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
for (int i = 0; i < numbers.size(); ++i) {
int low = i + 1, high = numbers.size() - 1; //low为i+1,防止重复元素
while (low <= high) {
int mid = low + (high - low) / 2;
if (numbers[mid] < target - numbers[i]) {
low = mid + 1;
}
else if (numbers[mid] == target - numbers[i]) {
return { i + 1, mid + 1 };
}
else {
high = mid - 1;
}
}
}
return { -1, -1 };
}
};
- 时间复杂度:O(nlogn),其中n是数组的长度,需要遍历数组确定第一个数,时间复杂度是O(n),寻找第二个数使用二分查找,时间复杂度是O(log n),因此总时间复杂度是O(nlogn)。
- 空间复杂度:O(1)。
方案二:双指针
定义low和high两个指针,分别从数组前面和后面开始遍历,将nums[low]和nums[high]之和 和目标值进行比较:(1)如果和小于目标值,则low指针向后移动;(2)如果和大于目标值,则high指针向前移动;(3)如果和等于目标值,则找到两数和等于目标值,返回对应下标组成的数组,此时应该注意,下标是从1开始,则对应下标需要都加上1。
//双指针
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
int low = 0, high = numbers.size() - 1;
while (low < high) { //不可以有等号,因为不可以重复使用同一元素
int sum = numbers[low] + numbers[high];
if (sum < target) {
low++;
}
else if (sum == target) {
return { low + 1, high + 1 };
}
else {
high--;
}
}
return { -1, -1 };
}
};
- 时间复杂度:O(n),n是数组的长度,每个元素最多被遍历一次。
- 空间复杂度:O(n)。