20.4.10 两数之和 II - 输入有序数组 简单
题目
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
解题思路
- 因为是有序,所以可以用二分法
- 定一个数作为index1,在这个数后面用二分法查找index2,找不到的话换一个index1,循环直到找到结果。
代码思路
- 循环遍历整个数组,将当前遍历的这个数定为index1,在index1后的数字用二分法寻找index2;
- 如果当前二分找不到合适的index2,更新index1;
- 若找到index2,标志flag变为true,跳出while和for循环,得出结果。
代码
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
vector<int> result;
if(numbers.size() == 0) return result;
int index1;
int index2;
for(int i = 0; i < numbers.size() - 1; i++){
index1 = i;
int begin = i + 1;
int end = numbers.size() - 1;
bool flag = false;
while(begin <= end){
int mid = (begin + end) / 2;
if(numbers[index1] + numbers[mid] == target){
flag = true;
index2 = mid;
break;
}
if(numbers[index1] + numbers[mid] > target) end = mid - 1;
if(numbers[index1] + numbers[mid] < target) begin = mid + 1;
}
if(flag) break;
}
result.push_back(index1 + 1);
result.push_back(index2 + 1);
return result;
}
};