题目:两数之和 II - 输入有序数组
描述:
给你一个下标从 1 开始的整数数组 numbers ,该数组已按 非递减顺序排列 ,请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] 和 numbers[index2] ,则 1 <= index1 < index2 <= numbers.length 。
以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 和 index2。
你可以假设每个输入 只对应唯一的答案 ,而且你 不可以 重复使用相同的元素。
你所设计的解决方案必须只使用常量级的额外空间。
示例 1:
输入:numbers = [2,7,11,15], target = 9
输出:[1,2]
解释:2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。
leetcode链接
方法一:双指针
由题目可知空间复杂度必须为常量级,暴力解时间复杂度为o(n²),经过测试时间超限,因此我们优化时间复杂度。题目给出的数组是有序的,因此我们考虑使用双指针的方法,定义两个指针left和right指针,分别指向头尾,如果这两个指针指向的数之和大于target,那么肯定需要往小的方向移动,那么就是right指针左边移动,反之大于target,肯定需要往大的数字移动,那么就是left指针右移动
时间复杂度:o(n),只遍历了一遍原数组
空间复杂度:o(1),没有申请额外的空间
vector<int> twoSum(vector<int>& numbers, int target) {
int n = numbers.size();
int left = 0,right = n-1;
vector<int> ans(2);
while(left<right){
if(numbers[left]+numbers[right]>target){
//大于目标数,right指针左移
right--;
}else if(numbers[left]+numbers[right]<target){
//小于目标数,left指针右边移
left++;
}else{
ans[0] = left+1;
ans[1] = right+1;
break;
}
}
return ans;
}
方法二:二分查找
考虑o(n²)的时间复杂度通过不了,我们考虑时间复杂度为o(nlog)的算法,那么和logn时间复杂度相关的算法就是二分查找,我们遍历数组,遍历到的当前元素为num1,如果num1<target,那么我们需要找的另外一个数就为target-num1,而我们利用数组有序的这一个性质,我们对第二个数去二分查找,查找的时间复杂度为logn,需要遍历一遍数组,所以总的时间复杂度为o(nlogn).
时间复杂度:o(nlogn)
空间复杂度:o(1)
vector<int> twoSum(vector<int>& numbers, int target) {
int n = numbers.size();
vector<int> ans(2);
for(int i=0;i<n;i++){
int left = i+1,right = n-1;
while(left<=right){
int mid = (left+right)/2;
if(numbers[mid]>target - numbers[i]){
right = mid - 1;
}else if(numbers[mid]<target - numbers[i]){
left = mid + 1;
}else{
return {i+1,mid+1};
}
}
}
return {-1,-1};
}