很久之前写过,今天在每日一题看到了,就重新写了一下,居然没有之前的方法快。。
最先想到的是时间复杂度O(N^2)的傻瓜式暴力,肯定超时,不说了。
今天想的是就遍历一遍,用map存每个数出现的下标,用set存每个数,然后对当前位置的数numbers[i],判断set里是否存在target-numbers[i],存在的话从map里取出下标,和 i+1 一起返回;不存在的话,map里放进当前数对应的下标,set放进当前的数。
class Solution {
public int[] twoSum(int[] numbers, int target) {
Map<Integer, Integer> map = new HashMap<>();
Set<Integer> set = new HashSet<>();
for(int i = 0; i < numbers.length; i++){
if(set.contains(target - numbers[i])){
return new int[]{map.get(target - numbers[i]), i + 1};
}else{
map.put(numbers[i], i + 1);
set.add(numbers[i]);
}
}
return null;
}
}
然后运行了一下,很慢。。虽然是O(N)的,但估计大部分时间浪费在了map和set的操作上。
回去看了下之前的写法,是双指针,因为原先数组已经排好序了,所以可以这么做。直接用left指向下标0,right指向最后一位,加起来等于target直接返回了,大于target右指针左移,小于target左指针右移,因为题目保证一定有解,所以不需要考虑特殊情况。
class Solution {
public int[] twoSum(int[] numbers, int target) {
int left = 0;
int right = numbers.length - 1;
while(left < right){
if(numbers[left] + numbers[right] == target){
return new int[]{left + 1, right + 1};
} else if(numbers[left] + numbers[right] > target){
right--;
} else{
left++;
}
}
return null;
}
}
可是这样还是没有最快?想了一下,while里面上面两个判断里对numbers数组的访问操作耗时多吗?那直接在while一进来先把这两个位置的数拿出来吧,空间换时间,终于1msl了。
class Solution {
public int[] twoSum(int[] numbers, int target) {
int left = 0;
int right = numbers.length - 1;
while(left < right){
int sum = numbers[left] + numbers[right];
if(sum == target){
return new int[]{left + 1, right + 1};
} else if(sum < target){
left++;
} else{
right--;
}
}
return null;
}
}