public class Solution {
public int[] twoSum(int[] numbers, int target) {
//还是采用之前TwoSum的解法,可以有O(N*N),或者O(N)+Map
//但这里提到了给定的数组是按升序排列的,那么是否可以利用这个来找到更好的解法?
int[] res = new int[2];
Map<Integer, Integer> map = new HashMap<>();//解法一:采用O(N)+Map
//因为解唯一,那么数组里的元素是唯一的
for(int i = 0; i < numbers.length; i++){
map.put(numbers[i], i + 1);//题中提到返回的index不是从0开始的
}
for(int i = 0; i < numbers.length; i++){
if(map.containsKey(target - numbers[i])){
res[0] = i + 1;
res[1] = map.get(target - numbers[i]);
return res;
}
}
return res;
}
}
解法二:在找target - numbers[i]时比用map要快,因为这里有个特殊情况是数组有序
public class Solution {
public int[] twoSum(int[] numbers, int target) {
//还是采用之前TwoSum的解法,可以有O(N*N),或者O(N)+Map
//但这里提到了给定的数组是按升序排列的,那么可以利用这个来找到更好的解法:
//既然涉及到了有序与查找,那么就可以采用二分查找(解法二)
int[] res = new int[2];
//题目中提到解是唯一的,说明数组中的元素没有重复
//而且存在解,则数组长度大于等于2
for(int i = 0; i < numbers.length - 1; i++){
int low = i + 1, high = numbers.length - 1;//不能用数组中的一个数两次,所以int low = i + 1
while(low <= high){//二分查找的固定格式,这里用到二分查找作用是同解法一的Map,找除了numbers[i]的另一个元素(毕竟二分查找是查一个数,不是查两个数)
int mid = (low + high) / 2;
if(numbers[mid] == target - numbers[i]){
res[0] = i + 1;
res[1] = mid + 1;
return res;
}
else if(numbers[mid] < target - numbers[i]){
low = mid + 1;//不是low++,注意二分查找的格式
}
else{
high = mid - 1;//不是low++,注意二分查找的格式
}
}
}
return res;
}
}
最优解(时间复杂度O(N)):
public class Solution {
public int[] twoSum(int[] numbers, int target) {
//还是采用之前TwoSum的解法,可以有O(N*N),或者O(N)+Map
//但这里提到了给定的数组是按升序排列的,那么可以利用这个来找到更好的解法:
//既然涉及到了有序,那么可以用两个指针分别从前、后找
int[] res = new int[2];
int left = 0, right = numbers.length - 1;
while(numbers[left] + numbers[right] != target){//不是left与right一个移一步,有可能一直是left或right在移
if(numbers[left] + numbers[right] < target){
left++;
}
else{
right--;
}
}
res[0] = left + 1;
res[1] = right + 1;
return res;
}
}