题目如下:
看到了一个巧妙的解法 时间复杂度小于 O(n2)
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] arr=new int[2];
HashMap<Integer,Integer> hashMap=new HashMap<Integer,Integer>();
//遍历每个数
for (int i = 0; i < nums.length; i++) {
//hashmap里都是互补的数,所以只要有一个数等于hashmap里的数就可以输出结果(因为只让输出一种结果)
if(hashMap.containsKey(nums[i])){
arr[1]=i;
arr[0]=hashMap.get(nums[i]);
return arr;
}
//用target减去当前的数,然后把减去后的数存入hashmap
hashMap.put(target-nums[i],i);
}
return arr;
}
}
那么我们将题目改一下,给的数组中的数都是有序的,那我们该怎么解呢?
第一种解法: hashmap
跟上面的一样,只不过效率较低
class Solution {
public int[] twoSum(int[] numbers, int target) {
HashMap<Integer,Integer> hashMap=new HashMap<>();
//遍历数组的每个数,不满足条件存进hashmap,满足条件输出
for(int i=0;i<numbers.length;i++){
if(hashMap.containsKey(numbers[i])) {
int a=hashMap.get(numbers[i]);
int b=i;
return new int[]{a,b};
}
hashMap.put(target-numbers[i],i);
}
return new int[0];
}
}
第二种解法:首位双指针,时间复杂度更低
解法来自评论区
由于数组递增的特性,所以后面的元素比前面的元素大
可以安排两个指针i,j在两头,记录两个指针的和
如果和大于target则代表需要缩小指针和,即需要右移j,来缩小指针和
如果和小于target则代表需要增大指针和,即需要左移i,来增大指针和
class Solution {
public int[] twoSum(int[] nums, int t) {
int len = nums.length-1;
for(int i = 0, j = len; i < j;){
int sum = nums[i] + nums[j];
if(sum > t) j--;
else if(sum < t) i++;
else return new int[]{i,j};
}
return new int[2];
}
}
总之,遇到有序,第一反应 二分和双指针