LeetCode 167. 两数之和2
给你一个下标从 1 开始的整数数组 numbers
,该数组已按 非递减顺序排列 ,请你从数组中找出满足相加之和等于目标数 target
的两个数。如果设这两个数分别是 numbers[index1]
和 numbers[index2]
,则 1 <= index1 < index2 <= numbers.length
。
以长度为 2 的整数数组 [index1, index2]
的形式返回这两个整数的下标 index1
和 index2
。
你可以假设每个输入 只对应唯一的答案 ,而且你 不可以 重复使用相同的元素。
你所设计的解决方案必须只使用常量级的额外空间。
输入:numbers=[2,7,11,15],target=9
输出:[1,2]
my_code
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
vector<int> a;
int ptr1=0,ptr2=numbers.size()-1;
while(1){
if(numbers[ptr1]+numbers[ptr2]==target) break;
if(numbers[ptr1]+numbers[ptr2]>target){
ptr2--;
}else if(numbers[ptr1]+numbers[ptr2]<target){
ptr1++;
}
}
a.push_back(ptr1+1);
a.push_back(ptr2+1);
return a;
}
};
思路:使用双指针,ptr1从小端开始,ptr2从大端开始。若和大于target,则大端左移,若和小于target,则小端右移。
- 时间复杂度O(n)
- 空间复杂度O(1)
方法二:二分查找
思路:在数组中找到两个数,使得它们的和等于目标值,可以首先固定第一个数,然后寻找第二个数,第二个数等于目标值减去第一个数的差。利用数组的有序性质,可以通过二分查找的方法寻找第二个数。为了避免重复寻找,在寻找第二个数时,只在第一个数的右侧寻找。
int* twoSum(int* numbers, int numbersSize, int target, int* returnSize) {
int* ret = (int*)malloc(sizeof(int) * 2);
*returnSize = 2;
for (int i = 0; i < numbersSize; ++i) {
int low = i + 1, high = numbersSize - 1;
while (low <= high) {
int mid = (high - low) / 2 + low;
if (numbers[mid] == target - numbers[i]) {
ret[0] = i + 1, ret[1] = mid + 1;
return ret;
} else if (numbers[mid] > target - numbers[i]) {
high = mid - 1;
} else {
low = mid + 1;
}
}
}
ret[0] = -1, ret[1] = -1;
return ret;
}
- 时间复杂度:O(nlogn)
- 空间复杂度:O(1)