题目
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明: 返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解答
方法一:
这个题目之前已经出现过了,唯一的不同就是返回的下标不同,需要进行把下标进行 +1操作。
这里展示一个头尾指针往中间移动的办法。分析如下:
- 升序排列。
- 先计算头尾指针指向的数据之和。
- 如果和大于target 那么尾指针向左移动那么和就会变小。
- 如果和小于target 那么头指针向右移动那么和就会变大。
- 直到找到和为target
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
//共同往中间移动
auto iter = numbers.begin();
auto riter = numbers.rbegin();
while (true) {
if (*iter + *riter == target) {
break;
} else if (*iter + *riter > target) {
riter++;
} else {
iter++;
}
}
int index1 = iter - numbers.begin() + 1;
int index2 = numbers.size() - (riter - numbers.rbegin());
return vector<int>{index1, index2};
}
};
这个代码虽然能跑通测试用例,但是使用 while(true)不太合理,因为如果不存在不存在target的话,指针会越界。
可以加上一个判断逻辑。
while (iter != numbers.end() && riter != numbers.rend()) {}