题干:
https://leetcode.cn/problems/two-sum-ii-input-array-is-sorted/
1、二分查找
在原版的两数之和问题(与本题同属一个专栏)中,我们把问题转化为查找问题,使用了哈希表查找。在这个问题中,由于已经是有序数组,我们可以直接对数组进行二分查找,效率也很高。
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
for (int i = 0; i < numbers.size(); i++) {
//lower_bound()二分查找返回值:指向首个不小于 value 的元素的迭代器,或若找不到这种元素则为 last 。
auto it = lower_bound(numbers.begin(), numbers.end(), target - numbers[i]);
//很有可能查找到了元素
if (it != numbers.end() && *it == target - numbers[i]) {
//这里是为了避免查找到自己
if ((it - numbers.begin()) != i) {
return vector<int>{i + 1, static_cast<int>(it - numbers.begin() + 1)};
}
else {
//查找到了自己,但是下一个可能是答案元素(有序数组)
if ( *it == *(it+1) ) {
return vector<int>{i + 1, static_cast<int>(it - numbers.begin() + 2)};
}
}
}
}
throw "don't find";
}
};
2、双指针遍历
有人妄图遍历一遍数组就找到符合的元素配对,笑话,这根本…有可能!
算法:两个指针分别指向头和尾,如果和大于target,那么尾指针左移,如果和小于target,那么头指针右移。直到找到答案退出,或者两个指针相等(意味着已经没有答案)退出。
可行性分析:假设有一对元素x,y符合条件,当我们的任意一个指针第一次指向了其中一个元素时x,另一个指针此时还没有指定到另一个答案元素y(这个指针没有指定到过答案元素y),由于数组是有序的,根据我们的算法,此时指定到x的指针不会移动,只有另一个指针不断移动逼近y(思考一下),最终使得我们找到答案。总之,双指针不会让我们错过答案。
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
int i{ 0 };//头指针(索引代替)
int j{ static_cast<int>( numbers.size() - 1 )};//尾指针
//i == j退出循环,此时代表找不到答案
while (i != j) {
if (numbers[i] + numbers[j] == target) {
return vector<int>{i + 1, j + 1};
}
else if (numbers[i] + numbers[j] > target) {
j--;
}
else if (numbers[i] + numbers[j] < target) {
i++;
}
}
throw "don't find";
}
};