这道题其实第一眼看到有序,(二分法的DNA就动了),但是题目只要求了常量级的额外空间,没有时间要求,所以我们可以通过逐个遍历,也就是 n² 的时间复杂度去完成这题目。
class Solution {
public int[] twoSum(int[] numbers, int target) {
for(int i=0;i<numbers.length;i++){
for(int j = i+1;j<numbers.length;j++){
if(numbers[i]+numbers[j]==target){
int[] res = new int[]{i+1,j+1};
return res;
}
}
}
return null;
}
}
时间复杂度太高了,不行,重写。有序,二分法,那我们是不是可以通过两个指针,第一个指针i指向第一个数,然后target - numbers[i] = x,把x作为二分查找的new target让他查找呢,这样或许对时间复杂度会有稍稍的提升。
说干就干:
class Solution {
public int[] twoSum(int[] numbers, int target) {
//二分,先固定一个点的位置,再去寻找
for(int i=0;i<numbers.length;i++){
//二分法实现寻找
int next = getNextIdx(i+1,numbers.length-1,numbers,target - numbers[i]);
if(next!=-999){
return new int[]{i+1,next+1};
}
}
return null;
}
public int getNextIdx(int startIdx,int endIdx,int[] numbers,int nextTarget){
while(startIdx<=endIdx){
int mid = startIdx + ((endIdx - startIdx)>>1);
if(numbers[mid]<nextTarget){
//中位数小于目标值,往右边找
startIdx = mid +1;
}else if(numbers[mid]>nextTarget){
//中位数大于目标值,往左找
endIdx = mid - 1;
}else{
return mid;
}
}
return -999;
}
}
再来看看时间
emmm,略有提升,但是还是比较高,因为还没有达到 常量级的时间复杂度,那常量级的时间复杂度怎么办呢。
根据上图,是我根据需要遍历的过程匹配过程用图的形式展现出来,其实可以看到,我们一开始取一个 最小 + 最大 ,也就是有序数组中的 number[0] + number[size-1]跟target比对,按照例题,target为7,number[0] + number[size-1] = 17 > 9,那不用说, number[i] + number[size -1] 全都是一定会 > 9 很明显不符合题目(因为i = 0 时已经是最小的数了) ,所以我们舍弃后续的遍历。
可以看到一次比对下来,把所有的15以及第一排的遍历都可以省略了,接下来是倒数第二个大的数字,也就是我们的11,
11 + 2 > 9,重复上述步骤
得到了答案。
可以看到,我们其实一共只是比对了3次,足以完成我们的题目,这个刚好符合常数级时间
int* twoSum(int* numbers, int numbersSize, int target, int* returnSize){
int* res = (int*)malloc(sizeof(int)*2);
memset(res,-1,sizeof(int)*2);
*returnSize = 2;
int i = 0;int j = numbersSize-1;
while(i<j){
//num[i] + num[j] < target ,证明头指针的数不够大,头指针后移
//num[i] + num[j] > target ,证明尾指针的数过大,尾指针前移
if(numbers[i] + numbers[j] > target){
j--;
}else if(numbers[i] + numbers[j] == target){
res[0] = i +1,res[1]=j+1;
return res;
}else{
i++;
}
}
return res ;
}
(然后不贴结果了,因为后面是用C写的,和java的没法对比)