题目描述:(来源leetcode167)
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
- 返回的下标值(index1 和 index2)不是从零开始的。
- 你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
1 暴力解法——双层遍历
这道题思路比较简单,通过两层遍历,就可以找出索引,需要注意边界的情况。下面实现一下:
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
vector<int> result;
for(int i=0;numbers[i]<=target;++i){
for(int j=i+1;j<numbers.size() && numbers[i]+numbers[j]<=target;++j){
if(numbers[i]+numbers[j]==target){
result.push_back(i+1);
result.push_back(j+1);
return result;
}
}
}
return result;
}
};
如果是在本地测试,当然算法实现没有问题,但是请注意上面的实现的时间复杂度是
O
(
n
2
)
O(n^2)
O(n2)的,当提交到leetcode上时,就会发现,超出时间限制
。
分析原因:
上面我们的暴力解法没有用到题目中的有序条件,一般我们提到有序,你应该联想到二分搜索树。二分搜索树可以将时间复杂度提高到
O
(
l
o
g
n
)
O(logn)
O(logn),在这道题中,我们也可以使用二分搜索树。
2 二分搜索树解法
思路:
1)目标数由两个元素组成,因此,可以确定其中一个元素,然后使用二分搜索法查找另一个元素;
2)注意边界,题中没有限定元素的正负,所以可以出现负数和0,这样就需要考虑边界等号问题;
class Solution {
public:
int binarySearch(vector<int>& arr,int l,int r,int target){
if(l>r)
return -1;//不存在时返回-1
int mid=l+(r-l)/2;
if(arr[mid]==target)
return mid+1;//题中要求返回索引从1开始
if(arr[mid]>target)
return binarySearch(arr,l,mid-1,target);
else
return binarySearch(arr,mid+1,r,target);
}
vector<int> twoSum(vector<int>& numbers, int target) {
vector<int> result;
int j=0;
for(int i=0;numbers[i]<=target;++i){
j=binarySearch(numbers,i+1,numbers.size()-1,target-numbers[i]);
if(j>-1){//只有存在j时,才将索引返回
result.push_back(i+1);
result.push_back(j);
return result;
}
}
return result;
}
};