题目
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
解题思路
遍历数组的元素,并用二分搜索尝试寻找和当前数相加为目标值的元素。同时主要要跳过重复选项。
C++解法
class Solution {
public:
int binarySearch(vector<int> numbers, int from, int to, int target) {
if (from > to) return -1;
int mid = (from + to) / 2;
if (target == numbers[mid]) return mid;
else if (target < numbers[mid]) return binarySearch(numbers, from, mid - 1, target);
else return binarySearch(numbers, mid + 1, to, target);
}
vector<int> twoSum(vector<int>& numbers, int target) {
int result = 0;
int n = (int)numbers.size();
for (int i = 0; i < numbers.size(); ++i) {
if (i > 0 && numbers[i] == numbers[i - 1]) continue;
if (numbers[i] > target) break;
if ((result = binarySearch(numbers, i + 1, n - 1, target - numbers[i])) != -1) {
return {i + 1, result + 1};
}
}
return {};
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted
网友评论